Fixed: Upgrade not working correctly.