diff --git a/pyproject.toml b/pyproject.toml
index a0c2e3eb..9e919604 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -55,6 +55,7 @@ rl = [
     "ray[rllib] >= 2.20.0, < 3",
     "tensorflow==2.12.0",
     "stable-baselines3[extra]==2.1.0",
+    "sb3-contrib==2.1.0",
 ]
 dev = [
     "build==0.10.0",