fixed ui bug for upgrades