Small update to reflect websocket redesign