Relearning Abilities W/ Upgrades

In a map I'm making, heros can learn non-hero abilities, unlearn them, and then learn them again. These skills level up though upgrades. The problem is when a hero relearns a skill the upgrade doesn't come with it and it's back to to level one. One of the things I've tried is using triggers to set the current level of the upgrade to the current level of the upgrade when the skill is learned, but it still doesn't work. Any suggestions?

You can't unlearn upgrades in any way. Find a different method to do the unit ability leveling.

variables of the spell level. when they relearn it apply variable.. id think that would be the best way to do it.

Thanks, I think I figured it out.