From aa0ceab4d3fb09bc431c66124daccb5409746ba9 Mon Sep 17 00:00:00 2001
From: Marks Polakovs <github@markspolakovs.me>
Date: Mon, 11 May 2020 02:20:06 +0200
Subject: [PATCH] Make sure we update the wsID, even on an existing
 registrationt

---
 stateserver.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/stateserver.py b/stateserver.py
index 73b58e3..3fb4830 100755
--- a/stateserver.py
+++ b/stateserver.py
@@ -276,6 +276,9 @@ def post_registerCheck() -> Any:
             # they've already registered, return the existing session
             print("found existing connection {} for {}".format(conn["connid"], conn["timeslotid"]))
             connection = conn
+            # make sure we update their wsID 
+            if "wsid" in content:
+                connection["wsid"] = content["wsid"]
 
     new_connection = False
     if connection is None: