Fix wrong update function being used