diff --git a/hidimstat/cpi.py b/hidimstat/cpi.py
index 8835fb8..815652f 100644
--- a/hidimstat/cpi.py
+++ b/hidimstat/cpi.py
@@ -1,10 +1,10 @@
 import numpy as np
 from joblib import Parallel, delayed
-from sklearn.base import BaseEstimator, TransformerMixin, check_is_fitted, clone
+from sklearn.base import BaseEstimator, check_is_fitted, clone
 from sklearn.metrics import root_mean_squared_error
 
 
-class CPI(BaseEstimator, TransformerMixin):
+class CPI(BaseEstimator):
     """
     Conditional Permutation Importance (CPI) algorithm.
     :footcite:t:`Chamma_NeurIPS2023` and for group-level see
diff --git a/hidimstat/loco.py b/hidimstat/loco.py
index e991469..6d691a3 100644
--- a/hidimstat/loco.py
+++ b/hidimstat/loco.py
@@ -1,10 +1,10 @@
 import numpy as np
 from joblib import Parallel, delayed
-from sklearn.base import BaseEstimator, TransformerMixin, check_is_fitted, clone
+from sklearn.base import BaseEstimator, check_is_fitted, clone
 from sklearn.metrics import root_mean_squared_error
 
 
-class LOCO(BaseEstimator, TransformerMixin):
+class LOCO(BaseEstimator):
     """
     Leave-One-Covariate-Out (LOCO) algorithm as described in
     :footcite:t:`Chamma_NeurIPS2023`.
diff --git a/hidimstat/permutation_importance.py b/hidimstat/permutation_importance.py
index 6010f71..f6fc496 100644
--- a/hidimstat/permutation_importance.py
+++ b/hidimstat/permutation_importance.py
@@ -1,10 +1,10 @@
 import numpy as np
 from joblib import Parallel, delayed
-from sklearn.base import BaseEstimator, TransformerMixin, check_is_fitted
+from sklearn.base import BaseEstimator, check_is_fitted
 from sklearn.metrics import root_mean_squared_error
 
 
-class PermutationImportance(BaseEstimator, TransformerMixin):
+class PermutationImportance(BaseEstimator):
     """