diff --git a/mifs/mifs.py b/mifs/mifs.py
index ecf074c..2543ace 100644
--- a/mifs/mifs.py
+++ b/mifs/mifs.py
@@ -11,7 +11,7 @@
 from sklearn.preprocessing import StandardScaler
 from multiprocessing import cpu_count
 from sklearn.base import BaseEstimator
-from sklearn.feature_selection.base import SelectorMixin
+from sklearn.feature_selection._base import SelectorMixin
 import bottleneck as bn
 from . import mi