fix(upgrades): callback is not triggering correctly