Change websocket behavior