diff --git a/src/primaite/main.py b/src/primaite/main.py
index 1211b5cd..c963dd00 100644
--- a/src/primaite/main.py
+++ b/src/primaite/main.py
@@ -25,6 +25,7 @@ from primaite.transactions.transactions_to_file import write_transaction_to_file
 def run_generic():
     """Run against a generic agent."""
     for episode in range(0, config_values.num_episodes):
+        env.reset()
         for step in range(0, config_values.num_steps):
             # Send the observation space to the agent to get an action
             # TEMP - random action for now
@@ -42,7 +43,6 @@ def run_generic():
             time.sleep(config_values.time_delay / 1000)
 
         # Reset the environment at the end of the episode
-        env.reset()
 
     env.close()
 
@@ -375,7 +375,7 @@ logging.info("Saving transaction logs...")
 
 write_transaction_to_file(transaction_list)
 
-config_file_main.close
+config_file_main.close()
 
 print("Finished")
 logging.info("Finished")