Fixing websocket