Update for WebSocket server mode