Partial Share Bug Fix (FAForever#4491)

speed2CZ · Dec 12, 2022 · 3b7d752 · 3b7d752
1 parent e4bf44c
commit 3b7d752
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/lua/SimUtils.lua b/lua/SimUtils.lua
@@ -19,11 +19,13 @@ local sharedUnits = {}
 function KillSharedUnits(owner, categoriesToKill)
     local sharedUnitOwner = sharedUnits[owner]
     if sharedUnitOwner and not table.empty(sharedUnitOwner) then
-        for _, unit in sharedUnitOwner do
+        local sharedUnitOwnerSize = table.getn(sharedUnitOwner)
+        for i = sharedUnitOwnerSize, 1, -1 do
+            local unit = sharedUnitOwner[i]
             if not unit.Dead and unit.oldowner == owner then
                 if categoriesToKill then
-                    if ContainsCategory(unit, categoriesToKill) then
-                        table.remove(sharedUnits[owner], unit)
+                    if EntityCategoryContains(categoriesToKill, unit) then
+                        table.remove(sharedUnits[owner], i)
                         unit:Kill()
                     end
                 else