From 76939fb8e80275789c8de868076d75c2934cfe67 Mon Sep 17 00:00:00 2001
From: Christopher McCarthy <chris.mccarthy@methods.co.uk>
Date: Thu, 23 Nov 2023 15:45:22 +0000
Subject: [PATCH] Apply suggestions from code review

---
 src/primaite/game/session.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/primaite/game/session.py b/src/primaite/game/session.py
index 572dbecb..a7df1161 100644
--- a/src/primaite/game/session.py
+++ b/src/primaite/game/session.py
@@ -281,7 +281,7 @@ class PrimaiteSession:
         self.episode_counter += 1
         self.step_counter = 0
         _LOGGER.debug(f"Restting primaite session, episode = {self.episode_counter}")
-        self.simulation = self._simulation_initial_state
+        self.simulation = copy.deepcopy(self._simulation_initial_state)
 
     def close(self) -> None:
         """Close the session, this will stop the env and close the simulation."""