diff --git a/builder/pwnagotchi.yml b/builder/pwnagotchi.yml
index 0740ec35..9a574aab 100644
--- a/builder/pwnagotchi.yml
+++ b/builder/pwnagotchi.yml
@@ -182,14 +182,11 @@
       regexp: "^CONF_SWAPSIZE=.*$"
       line: "CONF_SWAPSIZE=2048"
 
-  - name: update apt package cache
-    apt:
-      update_cache: yes
-
   - name: install packages
     apt:
       name: "{{ packages.apt.install }}"
       state: present
+      update_cache: yes
 
 # Install nexmon to fix wireless scanning (takes 2.5G of space)
   - name: clone nexmon repository
@@ -631,6 +628,10 @@
     apt:
       autoclean: true
 
+  - name: clean apt cache
+    apt:
+      clean: true
+
   - name: remove dependencies that are no longer required
     apt:
       autoremove: yes
diff --git a/requirements.txt b/requirements.txt
index decdc195..f5f2ad1d 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
-gym
+gymnasium
 shimmy
 pycryptodome
 requests