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