diff --git a/OpenRA.Mods.Cnc/Widgets/Logic/CncIngameChromeLogic.cs b/OpenRA.Mods.Cnc/Widgets/Logic/CncIngameChromeLogic.cs index 03224715c8..dad4807901 100644 --- a/OpenRA.Mods.Cnc/Widgets/Logic/CncIngameChromeLogic.cs +++ b/OpenRA.Mods.Cnc/Widgets/Logic/CncIngameChromeLogic.cs @@ -107,14 +107,16 @@ namespace OpenRA.Mods.Cnc.Widgets.Logic var sidebarRoot = playerWidgets.GetWidget("SIDEBAR_BACKGROUND"); - var sellButton = sidebarRoot.GetWidget("SELL_BUTTON"); + var sellButton = sidebarRoot.GetWidget("SELL_BUTTON"); sellButton.OnClick = () => world.ToggleInputMode(); + sellButton.IsToggled = () => world.OrderGenerator is SellOrderGenerator; var sellIcon = sellButton.GetWidget("ICON"); sellIcon.GetImageName = () => world.OrderGenerator is SellOrderGenerator ? "sell-active" : "sell"; - var repairButton = sidebarRoot.GetWidget("REPAIR_BUTTON"); + var repairButton = sidebarRoot.GetWidget("REPAIR_BUTTON"); repairButton.IsDisabled = () => !RepairOrderGenerator.PlayerIsAllowedToRepair( world ); repairButton.OnClick = () => world.ToggleInputMode(); + repairButton.IsToggled = () => world.OrderGenerator is RepairOrderGenerator; var repairIcon = repairButton.GetWidget("ICON"); repairIcon.GetImageName = () => repairButton.IsDisabled() ? "repair-disabled" : world.OrderGenerator is RepairOrderGenerator ? "repair-active" : "repair";