diff --git a/setup.cfg b/setup.cfg
index bde7f4dc..75ee26ee 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -3,7 +3,7 @@ line_length=100
 multi_line_output=4
 known_standard_library=numpy,setuptools
 known_myself=slowfast
-known_third_party=fvcore,iopath,av,torch,pycocotools,yacs,termcolor,scipy,simplejson,matplotlib,detectron2,torchvision,yaml,tqdm,psutil,opencv-python,pandas,tensorboard,moviepy,sklearn,cv2,PIL
+known_third_party=fvcore,iopath,av,torch,pycocotools,yacs,termcolor,scipy,simplejson,matplotlib,detectron2,torchvision,yaml,tqdm,psutil,opencv-python,pandas,tensorboard,moviepy,scikit-learn,cv2,Pillow
 no_lines_before=STDLIB,THIRDPARTY
 sections=FUTURE,STDLIB,THIRDPARTY,myself,FIRSTPARTY,LOCALFOLDER
 default_section=FIRSTPARTY
diff --git a/setup.py b/setup.py
index 185c7d64..c89c8090 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",
     ],