fix for can upgrade on units