minor tweak on upgrade