Websocket fix