diff --git a/replit_river/server.py b/replit_river/server.py
index 27f50fe..36eaabf 100644
--- a/replit_river/server.py
+++ b/replit_river/server.py
@@ -45,8 +45,7 @@ async def serve(self, websocket: WebSocketServerProtocol) -> None:
                 self._transport_options.session_disconnect_grace_ms / 1000,
             )
         except WebsocketClosedException:
-            # it is fine if the ws is closed during handshake, we just close the ws
-            await websocket.close()
+            # it is fine if the ws is closed during handshake, we just return
             return
         except Exception as e:
             logging.error(