Fix websocket communication issue