Fixed websocket issue