diff --git a/OpenRa.Game/UnitInfluenceMap.cs b/OpenRa.Game/UnitInfluenceMap.cs index 8be45447dc..9fad708018 100644 --- a/OpenRa.Game/UnitInfluenceMap.cs +++ b/OpenRa.Game/UnitInfluenceMap.cs @@ -57,8 +57,9 @@ namespace OpenRa.Game public void Remove( Actor self, IOccupySpace unit ) { - foreach (var c in unit.OccupiedCells()) - influence[c.X, c.Y] = null; + if (unit != null) + foreach (var c in unit.OccupiedCells()) + influence[c.X, c.Y] = null; } public void Update(Actor self, IOccupySpace unit)