diff --git a/pwnagotchi/ai/gym.py b/pwnagotchi/ai/gym.py
index 7134628b..6c0cac10 100644
--- a/pwnagotchi/ai/gym.py
+++ b/pwnagotchi/ai/gym.py
@@ -1,6 +1,6 @@
 import logging
-import gym
-from gym import spaces
+import gymnasium
+from gymnasium import spaces
 import numpy as np
 
 import pwnagotchi.ai.featurizer as featurizer
@@ -8,7 +8,7 @@ import pwnagotchi.ai.reward as reward
 from pwnagotchi.ai.parameter import Parameter
 
 
-class Environment(gym.Env):
+class Environment(gymnasium.Env):
     metadata = {'render.modes': ['human']}
     params = [
         Parameter('min_rssi', min_value=-200, max_value=-50),
diff --git a/pwnagotchi/ai/parameter.py b/pwnagotchi/ai/parameter.py
index 79f464c5..812e9a4d 100644
--- a/pwnagotchi/ai/parameter.py
+++ b/pwnagotchi/ai/parameter.py
@@ -1,4 +1,4 @@
-from gym import spaces
+from gymnasium import spaces
 
 
 class Parameter(object):
diff --git a/requirements.txt b/requirements.txt
index 757be904..08c417d8 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
-gym>=0.26.2
+Gymnasium
 shimmy
 pycryptodome>=3.9.4
 requests>=2.21.0