clear up aliveHealers also on UnitCreated, not only on UnitDestroyed

beyond-all-reason · May 18, 2024 · 41ff037 · 41ff037
1 parent 87d5341
commit 41ff037
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/luarules/gadgets/pve_areahealers.lua b/luarules/gadgets/pve_areahealers.lua
@@ -69,18 +69,21 @@ end
 
 function gadget:UnitCreated(unitID, unitDefID, unitTeam)
     if healersTable[unitDefID] then
+        Spring.Echo("Created Area Healer", unitID, UnitDefs[unitDefID].name)
         aliveHealers[unitID] = {
             healingpower = healersTable[unitDefID].healingpower,
             healingrange = healersTable[unitDefID].healingrange,
             canbehealed = healersTable[unitDefID].canbehealed,
         }
+    else
+        aliveHealers[unitID] = nil
     end
 end
 
 function gadget:UnitDestroyed(unitID, unitDefID, unitTeam, attackerID)
-    --if aliveHealers[unitID] then
+    if aliveHealers[unitID] then
         aliveHealers[unitID] = nil
-    --end
+    end
 end
 
 function gadget:GameFrame(frame)