diff --git a/OpenRA.Mods.Cnc/Traits/SupportPowers/ChronoshiftPower.cs b/OpenRA.Mods.Cnc/Traits/SupportPowers/ChronoshiftPower.cs index c30da5e12b..3dabafa4d6 100644 --- a/OpenRA.Mods.Cnc/Traits/SupportPowers/ChronoshiftPower.cs +++ b/OpenRA.Mods.Cnc/Traits/SupportPowers/ChronoshiftPower.cs @@ -192,8 +192,9 @@ namespace OpenRA.Mods.Cnc.Traits if (unit.CanBeViewedByPlayer(manager.Self.Owner)) { var decorations = unit.TraitsImplementing().FirstEnabledTraitOrDefault(); - foreach (var d in decorations.RenderSelectionAnnotations(unit, wr, Color.Red)) - yield return d; + if (decorations != null) + foreach (var d in decorations.RenderSelectionAnnotations(unit, wr, Color.Red)) + yield return d; } } } @@ -318,8 +319,9 @@ namespace OpenRA.Mods.Cnc.Traits if (unit.CanBeViewedByPlayer(manager.Self.Owner)) { var decorations = unit.TraitsImplementing().FirstEnabledTraitOrDefault(); - foreach (var d in decorations.RenderSelectionAnnotations(unit, wr, Color.Red)) - yield return d; + if (decorations != null) + foreach (var d in decorations.RenderSelectionAnnotations(unit, wr, Color.Red)) + yield return d; } } } diff --git a/OpenRA.Mods.Common/Traits/SupportPowers/GrantExternalConditionPower.cs b/OpenRA.Mods.Common/Traits/SupportPowers/GrantExternalConditionPower.cs index 64f28ad94b..8540234a09 100644 --- a/OpenRA.Mods.Common/Traits/SupportPowers/GrantExternalConditionPower.cs +++ b/OpenRA.Mods.Common/Traits/SupportPowers/GrantExternalConditionPower.cs @@ -155,8 +155,9 @@ namespace OpenRA.Mods.Common.Traits foreach (var unit in power.UnitsInRange(xy)) { var decorations = unit.TraitsImplementing().FirstEnabledTraitOrDefault(); - foreach (var d in decorations.RenderSelectionAnnotations(unit, wr, Color.Red)) - yield return d; + if (decorations != null) + foreach (var d in decorations.RenderSelectionAnnotations(unit, wr, Color.Red)) + yield return d; } }