I agree with you! The dungeon already does this, I will unify it for the next update so your units can upgrade as you progress.
