Update WebSocket.php