diff --git a/store/python/test.py b/store/python/test.py
index f300392..2157e69 100644
--- a/store/python/test.py
+++ b/store/python/test.py
@@ -1,6 +1,7 @@
-import trictrac
+import store
+# import trictrac
 
-game = trictrac.TricTrac()
+game = store.TricTrac()
 print(game.get_state())  # "Initial state"
 
 moves = game.get_available_moves()
diff --git a/store/python/trictracEnv.py b/store/python/trictracEnv.py
index 4e40e33..f4524b2 100644
--- a/store/python/trictracEnv.py
+++ b/store/python/trictracEnv.py
@@ -1,7 +1,8 @@
 import gymnasium as gym
 import numpy as np
 from gymnasium import spaces
-import trictrac  # module Rust exposé via PyO3
+# import trictrac  # module Rust exposé via PyO3
+import store  # module Rust exposé via PyO3
 from typing import Dict, List, Tuple, Optional, Any, Union
 
 class TricTracEnv(gym.Env):
@@ -13,7 +14,7 @@ class TricTracEnv(gym.Env):
         super(TricTracEnv, self).__init__()
 
         # Instancier le jeu
-        self.game = trictrac.TricTrac()
+        self.game = store.TricTrac()
 
         # Stratégie de l'adversaire
         self.opponent_strategy = opponent_strategy