diff --git a/setup.py b/setup.py
index c0689e3..058b8ca 100644
--- a/setup.py
+++ b/setup.py
@@ -17,7 +17,7 @@
         "opencv-python",
         "torchvision>=0.4.2",
         "Pillow",
-        "sklearn",
+        "scikit-learn",
         "fairscale",
     ],
     packages=find_packages(exclude=("configs", "tests")),