small fix on upgrade class