Fix auto upgrades