diff --git a/zoo/policies/keyboard_agent.py b/zoo/policies/keyboard_agent.py
index 70c8dae8f8..58d0b0132c 100644
--- a/zoo/policies/keyboard_agent.py
+++ b/zoo/policies/keyboard_agent.py
@@ -1,6 +1,6 @@
 from typing import Callable, Tuple
 
-import gym
+import gymnasium as gym
 import numpy as np
 from pynput import keyboard