diff --git a/requirements.txt b/requirements.txt
index 9918d7e..05c3535 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -22,5 +22,5 @@ dbus-python==1.2.12
 toml==0.10.0
 python-dateutil==2.8.1
 websockets==8.1
-stable-baselines[mpi]==2.10.0
+stable-baselines3
 RPi.GPIO