diff --git a/gtep/driver_coal.py b/gtep/driver_coal.py
index 54fc0a3..b8ccf79 100644
--- a/gtep/driver_coal.py
+++ b/gtep/driver_coal.py
@@ -48,7 +48,7 @@
 # 4 representative days (1 per season)
 # Hourly commitment and dispatch
 mod_object = ExpansionPlanningModel(
-    stages=3, data=data_object, num_reps=5, len_reps=24, num_commit=24, num_dispatch=1
+    stages=3, data=data_object, num_reps=4, len_reps=24, num_commit=24, num_dispatch=1
 )
 # print(mod_object.data.data["elements"]["generator"]["1"])
 # import sys
@@ -128,9 +128,9 @@
 costs = {}
 for exp in mod_object.model.component_objects(pyo.Expression, descend_into=True):
     if "operatingCost" in exp.name:
-        costs[var.name] = pyo.value(exp)
+        costs[exp.name] = pyo.value(exp)
     elif "investmentCost" in exp.name:
-        costs[var.name] = pyo.value(exp)
+        costs[exp.name] = pyo.value(exp)
 
 import json