Fix demo with WebSocket