From 73556d37ec20b2fb238861d6a56f33cf1ff86ce0 Mon Sep 17 00:00:00 2001
From: Frank Villaro-Dixon <frank@villaro-dixon.eu>
Date: Mon, 2 Sep 2024 12:07:33 +0200
Subject: [PATCH] fix cleanup

Signed-off-by: Frank Villaro-Dixon <frank@villaro-dixon.eu>
---
 proxy.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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