fix websocket remove bug