Removing websocket argument