fix(upgrades): timeout callback is not triggering correctly