Websocket error handling fixes