diff --git a/setup.py b/setup.py
index 9ca9a05..0d97d95 100644
--- a/setup.py
+++ b/setup.py
@@ -19,4 +19,5 @@ setup(
             'pandas',
             'ndsplines',
             'timeout_decorator',
-            'tqdm'])
+            'tqdm',
+            'gymnasium'])