Perhaps we could add mana cost per level to the ability, something along those lines.
Maybe... It would just make the ability more universal. Since fire depends on magic, and cold depends on the creature's damage. Then it could be given to both mages and warriors, because fire is random, while cold is constant...