change onUpdate callback behaviour