WebSocket fixes