diff --git a/gym_tic_tac_toe/envs/tic_tac_toe_env.py b/gym_tic_tac_toe/envs/tic_tac_toe_env.py
index 8af4384..5c374aa 100644
--- a/gym_tic_tac_toe/envs/tic_tac_toe_env.py
+++ b/gym_tic_tac_toe/envs/tic_tac_toe_env.py
@@ -55,8 +55,8 @@ class TicTacToeEnv(gym.Env):
     def move_generator(self):
         moves = []
         for i in range (9):
-            if (self.state.state['board'][i]== 0):
-                p = self.state.on_move
+            if (self.state['board'][i]== 0):
+                p = self.state['on_move']
                 m = [p, i]
                 moves.append(m)