From 3760ac2811da8a10c5b77bfaf0e977cd9a394749 Mon Sep 17 00:00:00 2001
From: lL1l1 <82986251+lL1l1@users.noreply.github.com>
Date: Tue, 13 Feb 2024 01:06:18 -0800
Subject: [PATCH] Allow the upgrade hotkey to upgrade everything

---
 lua/keymap/hotbuild.lua | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lua/keymap/hotbuild.lua b/lua/keymap/hotbuild.lua
index b36d87f116..7f678c4bd2 100644
--- a/lua/keymap/hotbuild.lua
+++ b/lua/keymap/hotbuild.lua
@@ -615,11 +615,11 @@ end
 -- Does support upgrading T1 structures (facs, radars, etc.) that are currently upgrading to T2 to T3 when issued
 function buildActionUpgrade()
     local selectedUnits = GetSelectedUnits()
-    local availableOrders, availableToggles, buildableCategories = GetUnitCommandData(selectedUnits)
     local result = true
 
     for index, unit in selectedUnits do
         local bp = unit:GetBlueprint()
+        local _, _, buildableCategories = GetUnitCommandDataOfUnit(unit)
         -- If upgradeTab[bp.BlueprintId] returns a table, there are two or more possible upgrades, e.g. HQ and
         -- support factory. If instead bp.General.UpgradesTo returns a string, then there is only one possible
         -- upgrade, e.g. for a radar. For our helper function IssueUpgradeCommand we want a table, hence the { }