Fix websocket