diff --git a/setup.py b/setup.py
index 2b3ab1a0472b10b67f9c98457541a1795d16aa67..edafc1b771c7bd4db8923092080d7252869d3deb 100644
--- a/setup.py
+++ b/setup.py
@@ -25,7 +25,8 @@ def setup_package():
                       'scikit-learn>=0.19.0', 'scipy>=1.3.0', 'six>=1.12.0',
                       'pandas>=0.23.3', '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"],
+                      "randomscm @ git+https://github.com/thibgo/randomscm.git#egg=randomscm",
+                      "imbalanced-learn"],
     extras_require={
             'dev': ['pytest', 'pytest-cov'],
             'doc': ['sphinx >= 3.0.2', 'numpydoc', 'docutils', 'sphinx-autoapi',
diff --git a/summit/tests/test_multi_view/test_multiview_utils.py b/summit/tests/test_multi_view/test_multiview_utils.py
index 711b152d3bd426f4335a621d98c1ae8751ec7263..3fca7672456e440a7cf6739f13841135177af68e 100644
--- a/summit/tests/test_multi_view/test_multiview_utils.py
+++ b/summit/tests/test_multi_view/test_multiview_utils.py
@@ -76,6 +76,7 @@ class TestFunctions(unittest.TestCase):
         self.assertEqual(avail, ['adaboost',
                                  'decision_tree',
                                  'gradient_boosting',
+                                 'imbalance_bagging',
                                  'knn',
                                  'lasso',
                                  'random_forest',
@@ -90,6 +91,7 @@ class TestFunctions(unittest.TestCase):
         self.assertEqual(avail, ['adaboost',
                                  'decision_tree',
                                  'gradient_boosting',
+                                 'imbalance_bagging',
                                  'knn',
                                  'random_forest',
                                  "random_scm",