This change should probably be backported to LH as well
Could it be changed to the last unit in a stack doesn't take all the damage? Damage should first go to the tankiest unit, at least from melee units. Ranged units should take damage from ranged/magic.