diff --git a/proxy.py b/proxy.py
index 7504966..7aae5ac 100755
--- a/proxy.py
+++ b/proxy.py
@@ -29,7 +29,7 @@ async def handler(websocket):
         print(f"Connection closed: {e}")
     finally:
         # Unregister the client
-        connected_clients.remove(websocket)
+        connected_clients.get(websocket.request.path, set()).remove(websocket)
 
 async def main():
     # Start the WebSocket server