Changing Attack Type Upgrade

I searched the forum, and didn't find an answer..so here goes.

I know you are able to create an upgrade that changes the Defense Type from Unarmored ---> Divine or whatever, but how would I create an upgrade that changes the attack? Specifically, Chaos.

You can't directly use an upgrade to change attack types. The simplest thing to do i think would be something like this:

Have the unit with say, normal attack type set for attack 1. Then make attack 2 the exact same except for the attack type. Then set the enabled attacks to attack one only for the unit.

Then create an upgrade that has Enable Attacks, and set attacks enabled to "attack two only".

Use an orb that changes its attack to the type you want and give the unit that orb ability when the upgrade is researched.

There is another way..
1.Create a clone of the unit with chaos(or whatever you want)
2.Create a clone of the chaos ability(and change the unit thing to the clone of the unit you want to be changed)
3.Create a new uppgrade
4.Fix that the ability requires the new Uppgrade
5.Test the map and try to uppgrade, the "normal" unit will now be changed to the "custom" unit.

(i'm sure there is an easier way.. ..but do wahtever you want)

The defense type change upgrade doesn't help..it either changes it to unarmored when I put Divine/Hero, or it doesn't work at all.