Websocket fixes