Adding upgrade_in_progress at the gear level