Fix on Upgrade command.