upgrade button bug fixes