Fix upgrade button