diff --git a/setup.py b/setup.py
index 7314149..7c38c62 100644
--- a/setup.py
+++ b/setup.py
@@ -22,7 +22,7 @@
         'torch_cluster',
         'tqdm',
         'numpy',
-        'sklearn',
+        'scikit-learn',
         'atom3d'
     ]
-)
\ No newline at end of file
+)