fix websocket bug