fixed bug of units not showing up on upgrade