websocket little code change