diff --git a/setup.py b/setup.py
index 47d99cf..14a77f8 100644
--- a/setup.py
+++ b/setup.py
@@ -11,7 +11,7 @@
       install_requires=[
           'tensorflow-gpu',
           'torch',
-          'sklearn',
+          'scikit-learn',
           'transformers==2.3.0', 'keras', 'pandas', 'numpy'
       ],