diff --git a/OpenRA.Mods.Cnc/Traits/Disguise.cs b/OpenRA.Mods.Cnc/Traits/Disguise.cs index 939b00d174..34eabacee1 100644 --- a/OpenRA.Mods.Cnc/Traits/Disguise.cs +++ b/OpenRA.Mods.Cnc/Traits/Disguise.cs @@ -25,16 +25,15 @@ namespace OpenRA.Mods.Cnc.Traits public override object Create(ActorInitializer init) { return new DisguiseTooltip(init.Self, this); } } - class DisguiseTooltip : ITooltip + class DisguiseTooltip : ConditionalTrait, ITooltip { readonly Actor self; readonly Disguise disguise; - TooltipInfo info; - public DisguiseTooltip(Actor self, TooltipInfo info) + public DisguiseTooltip(Actor self, DisguiseTooltipInfo info) + : base(info) { this.self = self; - this.info = info; disguise = self.Trait(); } @@ -42,7 +41,7 @@ namespace OpenRA.Mods.Cnc.Traits { get { - return disguise.Disguised ? disguise.AsTooltipInfo : info; + return disguise.Disguised ? disguise.AsTooltipInfo : Info; } }