diff --git a/web/app.py b/web/app.py
index 4dc6e75..bdae264 100644
--- a/web/app.py
+++ b/web/app.py
@@ -88,7 +88,11 @@ class _WSInterceptMiddleware:
 
     async def __call__(self, scope, receive, send) -> None:
         if scope["type"] == "websocket" and scope.get("path") == "/ws":
-            await self._ws(scope, receive, send)
+            from urllib.parse import parse_qs
+            from starlette.websockets import WebSocket as _WS
+            qs = parse_qs(scope.get("query_string", b"").decode())
+            token = qs.get("token", [""])[0]
+            await self._ws(_WS(scope, receive=receive, send=send), token)
         else:
             await self._app(scope, receive, send)