WebSocket fix