Websocket client fix