diff --git a/setup.py b/setup.py
index 8bb833cb9..0dcb7c8d5 100644
--- a/setup.py
+++ b/setup.py
@@ -4,7 +4,7 @@
 
 install_requires = [
     'torch>=1.3.0',
-    'gymnasium[all]',
+    'gymnasium[atari]',
     'numpy>=1.11.0',
     'pillow',
     'filelock',