diff --git a/setup.py b/setup.py
index b78cfefd4b0028b3b13e538d04c14f35ea403381..78f712b323e825fb84325d2031464e6056fb53d2 100644
--- a/setup.py
+++ b/setup.py
@@ -26,7 +26,7 @@ def setup_package():
                       'pandas==0.23', 'pyyaml>=3.12', 'plotly>=4.2.1',
                       'matplotlib==3.1.1', 'tabulate==0.8.6', 'pyscm-ml==1.0.0',
                       "randomscm @ git+https://github.com/thibgo/randomscm.git#egg=randomscm",
-                      "imblearn==0.9.1"],
+                      "imblearn==0.9"],
     extras_require={
             'dev': ['pytest', 'pytest-cov'],
             'doc': ['sphinx >= 3.0.2', 'numpydoc', 'docutils', 'sphinx-autoapi',