diff --git a/eos/saveddata/fit.py b/eos/saveddata/fit.py index 3f14026710..f49c1f60e9 100644 --- a/eos/saveddata/fit.py +++ b/eos/saveddata/fit.py @@ -737,7 +737,7 @@ def __generateDrain(self): capAdded = 0 for mod in self.modules: if mod.state >= State.ACTIVE: - if mod.getModifiedItemAttr("capacitorNeed") != 0 and mod.getModifiedItemAttr("capacitorNeed") is not None: + if (mod.getModifiedItemAttr("capacitorNeed") or 0) != 0: cycleTime = mod.rawCycleTime or 0 reactivationTime = mod.getModifiedItemAttr("moduleReactivationDelay") or 0 fullCycleTime = cycleTime + reactivationTime