Fix bug (websocket.js)