Update websocket controller