Add missing update calls