Milita, changes to Peasant when loaded into a transport, how to stop that?

[CaptainHero]

Milita are very hard coded, i know..
However I can work with them so far, but since i wanted to make the effect permament, its stupid that you can abort the effect by simply putting the unit into a Transport, since then they will just become peasnt by default... and why would you run back to the Town Hall if you could just jump in and out of the next transport?
Soo any solutions?

 

[Lender]

well, trigger the ability by yourself:/ compleatly replace the unit with an other with militia model.
that should work.

 

[Vinz]

Probably a custom Call To Arms, but it's too much work.