From dfe07ad92650c63d7478eea73881d9d88c35b4b0 Mon Sep 17 00:00:00 2001
From: Chris McCarthy <chris.mccarthy@methods.co.uk>
Date: Thu, 23 Nov 2023 21:25:52 +0000
Subject: [PATCH] #2034 - Fixed deepcopy reference issue

---
 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 a7df1161..1167f3d4 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 = copy.deepcopy(self._simulation_initial_state)
+        self.simulation = deepcopy(self._simulation_initial_state)
 
     def close(self) -> None:
         """Close the session, this will stop the env and close the simulation."""