diff --git a/rts/Sim/Units/Unit.cpp b/rts/Sim/Units/Unit.cpp index e8b29b5500..38719b38e2 100644 --- a/rts/Sim/Units/Unit.cpp +++ b/rts/Sim/Units/Unit.cpp @@ -8,6 +8,7 @@ #include "UnitMemPool.h" #include "UnitToolTipMap.hpp" #include "UnitTypes/Building.h" +#include "UnitTypes/ExtractorBuilding.h" #include "Scripts/NullUnitScript.h" #include "Scripts/UnitScriptFactory.h" #include "Scripts/CobInstance.h" // for TAANG2RAD @@ -1981,6 +1982,12 @@ bool CUnit::AddBuildPower(CUnit* builder, float amount) beingBuilt = true; SetMetalStorage(0); SetEnergyStorage(0); + + // make sure neighbor extractors update + CExtractorBuilding* extractor = dynamic_cast(this); + if (extractor != nullptr) + extractor->ResetExtraction(); + eventHandler.UnitReverseBuilt(this); }