private void OnUnitDied(UnitDiedEvent customEvent)
{
UnitStackBase unit = customEvent.Unit;
if (playerUnitsInBattle.Contains(unit))
{
playerUnitsInBattle.Remove(unit);
}
else if (enemyUnitsInBattle.Contains(unit))
{
enemyUnitsInBattle.Remove(unit);
}
else
{
return;
}
unit.gameObject.SetActive(false);
deadUnits.Add(unit);
battleUI.DeselectCurrentTarget();
}
