Fix WebSocket