feat: add when condition to allow other upgrades in the future