fix for upgrades