diff --git a/setup.py b/setup.py
index 34bc88b..c5b5c5d 100644
--- a/setup.py
+++ b/setup.py
@@ -23,7 +23,7 @@
     "stable-baselines3": ["stable-baselines3[extra]~=2.1.0", "pyyaml"],
     "ray-rllib": ["ray[rllib]~=2.7.0", "tensorflow", "torch", "pyyaml"],
     "sheeprl": [
-        "sheeprl==0.5.1",
+        "sheeprl==0.5.4",
         "importlib-resources==6.1.0",
     ],
 }