diff --git a/gtep/model_library/dispatch.py b/gtep/model_library/dispatch.py
index 166334c..dd0156c 100644
--- a/gtep/model_library/dispatch.py
+++ b/gtep/model_library/dispatch.py
@@ -228,7 +228,7 @@ def CP_flow_balance(b):
             buses = [bus for bus in m.buses]
             loads = [l for l in b.loads]
             gens = [gen for gen in m.generators]
-            batts = [bat for bat in m.storage]
+            batts = [bat for bat in m.storage] if m.config["storage"] else []
             balance += sum(
                 b.thermalGeneration[g] for g in gens if g in m.thermalGenerators
             )