Tweak on Upgrade Command.