upgradebutton, removebutton fix