Fix the WebSocketClient example