Fixed the upgrades