diff --git a/setup.py b/setup.py
index 185c7d64..33d3f8ff 100644
--- a/setup.py
+++ b/setup.py
@@ -23,8 +23,8 @@
         "opencv-python",
         "pandas",
         "torchvision>=0.4.2",
-        "PIL",
-        "sklearn",
+        "pillow",
+        "scikit-learn",
         "tensorboard",
         "fairscale",
     ],