diff --git a/flower_game_env.py b/flower_game_env.py
index b66bef8..38fa0d7 100644
--- a/flower_game_env.py
+++ b/flower_game_env.py
@@ -226,14 +226,14 @@ class FlowerGameEnv(gym.Env):
         return obs, {}
 
     def step(self, action):
-        # if action == 0:
-        #     pyautogui.press("w")
-        # elif action == 1:
-        #     pyautogui.press("a")
-        # elif action == 2:
-        #     pyautogui.press("s")
-        # elif action == 3:
-        #     pyautogui.press("d")
+        if action == 0:
+            pyautogui.press("w")
+        elif action == 1:
+            pyautogui.press("a")
+        elif action == 2:
+            pyautogui.press("s")
+        elif action == 3:
+            pyautogui.press("d")
 
         time.sleep(0.01)