Let’s come up with a monster.

Could it be possible to implement a feature where both the unupgraded unit and the upgraded unit gain the ability to dig pits? The upgraded unit could then gain the additional ability to dig a pit next to a friendly unit (without affecting the pit's movement range), allowing you to protect a ranged unit from melee units.