fix: closes #716

server.py

@@ -52,19 +52,20 @@ async def broadcast_handler(websocket: websockets.WebSocketServerProtocol):
                 continue
                 
             disconnected = set()
-            # Create broadcast tasks for efficiency
-            tasks = []
+            # Create broadcast tasks, tracking which client each task targets
+            task_client_pairs = []
             for client in clients:
                 if client != websocket and client.open:
-                    tasks.append(asyncio.create_task(client.send(message)))
-            
-            if tasks:
+                    task = asyncio.create_task(client.send(message))
+                    task_client_pairs.append((task, client))
+
+            if task_client_pairs:
+                tasks = [pair[0] for pair in task_client_pairs]
                 results = await asyncio.gather(*tasks, return_exceptions=True)
                 for i, result in enumerate(results):
                     if isinstance(result, Exception):
-                        # Find the client that failed
-                        target_client = [c for c in clients if c != websocket][i]
-                        logger.error(f"Failed to send to a client {target_client.remote_address}: {result}")
+                        target_client = task_client_pairs[i][1]
+                        logger.error(f"Failed to send to client {target_client.remote_address}: {result}")
                         disconnected.add(target_client)
             
             if disconnected: