Websocket fix 2