diff --git a/simulation-system/libs/gym-csle-stopping-game/README.MD b/simulation-system/libs/gym-csle-stopping-game/README.MD
index 1e0e338b1..92d563b2f 100644
--- a/simulation-system/libs/gym-csle-stopping-game/README.MD
+++ b/simulation-system/libs/gym-csle-stopping-game/README.MD
@@ -15,6 +15,7 @@ An OpenAI gym reinforcement learning environment that represents the optimal sto
 - `csle-attacker`
 - `csle-defender`
 - `csle-collector`
+- `csle-system-identification`
 - `gymnasium` for specifying the RL environment API
 
 ## Development Requirements
diff --git a/simulation-system/libs/gym-csle-stopping-game/requirements.txt b/simulation-system/libs/gym-csle-stopping-game/requirements.txt
index 8da60745d..2ab11df47 100644
--- a/simulation-system/libs/gym-csle-stopping-game/requirements.txt
+++ b/simulation-system/libs/gym-csle-stopping-game/requirements.txt
@@ -3,4 +3,5 @@ csle-common==0.6.2
 csle-collector==0.6.2
 csle-attacker==0.6.2
 csle-defender==0.6.2
+csle-system-identification==0.6.2
 gymnasium==0.27.1
\ No newline at end of file
diff --git a/simulation-system/libs/gym-csle-stopping-game/setup.cfg b/simulation-system/libs/gym-csle-stopping-game/setup.cfg
index 30e7740c2..1bf618cb1 100644
--- a/simulation-system/libs/gym-csle-stopping-game/setup.cfg
+++ b/simulation-system/libs/gym-csle-stopping-game/setup.cfg
@@ -25,6 +25,7 @@ install_requires =
     csle-attacker>=0.6.2
     csle-defender>=0.6.2
     csle-collector>=0.6.2
+    csle-system-identification>=0.6.2
 python_requires = >=3.8
 package_dir =
     =src
diff --git a/simulation-system/libs/gym-csle-stopping-game/tests/test_stopping_game_env.py b/simulation-system/libs/gym-csle-stopping-game/tests/test_stopping_game_env.py
index eef079c70..414c35d19 100644
--- a/simulation-system/libs/gym-csle-stopping-game/tests/test_stopping_game_env.py
+++ b/simulation-system/libs/gym-csle-stopping-game/tests/test_stopping_game_env.py
@@ -1,7 +1,7 @@
 from typing import Dict, Any
 import pytest
 from unittest.mock import patch, MagicMock
-from gym.spaces import Box, Discrete
+from gymnasium.spaces import Box, Discrete
 import numpy as np
 from gym_csle_stopping_game.envs.stopping_game_env import StoppingGameEnv
 from gym_csle_stopping_game.dao.stopping_game_config import StoppingGameConfig