From 5e83cfbbcfcc4cf21994b9c868bde338f438d20e Mon Sep 17 00:00:00 2001
From: Jose Daniel Lara <jdlara@berkeley.edu>
Date: Fri, 29 Sep 2023 01:15:02 -0600
Subject: [PATCH] fix columns call

---
 src/simulation/simulation_state.jl | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/simulation/simulation_state.jl b/src/simulation/simulation_state.jl
index e628294e89..b2f4a79d4c 100644
--- a/src/simulation/simulation_state.jl
+++ b/src/simulation/simulation_state.jl
@@ -148,27 +148,27 @@ function _initialize_system_states!(
                 make_system_state(
                     simulation_initial_time,
                     min_res,
-                    cols,
+                    column_names,
                 ),
             )
         end
     end
 
     for key in get_dataset_keys(decision_states)
+        dm_cols = get_column_names(key, get_dataset(decision_states, key))
         if has_dataset(emulator_states, key)
-            dm_cols = get_column_names(key, get_dataset(decision_states, key))
             em_cols = get_column_names(key, get_dataset(emulator_states, key))
             @assert_op dm_cols == em_cols
             continue
         end
-        cols = get_column_names(key, get_dataset(decision_states, key))
+
         set_dataset!(
             emulator_states,
             key,
             make_system_state(
                 simulation_initial_time,
                 min_res,
-                cols,
+                dm_cols,
             ),
         )
     end