diff --git a/setup.py b/setup.py
index 6d3da28..903ce81 100644
--- a/setup.py
+++ b/setup.py
@@ -29,7 +29,7 @@
     install_requires=[
         'numpy >= 1.11.1',
         'pandas',
-        'sklearn',
+        'scikit-learn',
         'tensorflow',
         'tensorflow-probability',
         'tqdm',