Fix websocket updates