diff --git a/builder/pwnagotchi.yml b/builder/pwnagotchi.yml
index 167c71a4..8368dc25 100644
--- a/builder/pwnagotchi.yml
+++ b/builder/pwnagotchi.yml
@@ -90,6 +90,7 @@
           - libbz2-dev
           - libc-ares-dev
           - libc6-dev
+          - libcap-dev
           - libdbus-1-dev
           - libdbus-glib-1-dev
           - libeigen3-dev
diff --git a/pwnagotchi/ai/gym.py b/pwnagotchi/ai/gym.py
index 6c0cac10..7134628b 100644
--- a/pwnagotchi/ai/gym.py
+++ b/pwnagotchi/ai/gym.py
@@ -1,6 +1,6 @@
 import logging
-import gymnasium
-from gymnasium import spaces
+import gym
+from gym 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(gymnasium.Env):
+class Environment(gym.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 45e9a85a..414129b7 100644
--- a/pwnagotchi/ai/parameter.py
+++ b/pwnagotchi/ai/parameter.py
@@ -1,4 +1,4 @@
-from gymnasium import spaces
+from gym import spaces
 
 
 class Parameter(object):
diff --git a/requirements.txt b/requirements.txt
index f5f2ad1d..decdc195 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
-gymnasium
+gym
 shimmy
 pycryptodome
 requests