diff --git a/mambular/models/sklearn_base_classifier.py b/mambular/models/sklearn_base_classifier.py
index 1149e14d..3188a578 100644
--- a/mambular/models/sklearn_base_classifier.py
+++ b/mambular/models/sklearn_base_classifier.py
@@ -27,6 +27,7 @@ class SklearnBaseClassifier(BaseEstimator):
     def __init__(self, model, config, **kwargs):
         self.preprocessor_arg_names = [
             "n_bins",
+            "feature_preprocessing",
             "numerical_preprocessing",
             "categorical_preprocessing",
             "use_decision_tree_bins",
diff --git a/mambular/models/sklearn_base_lss.py b/mambular/models/sklearn_base_lss.py
index e1b63a74..cfb8d17e 100644
--- a/mambular/models/sklearn_base_lss.py
+++ b/mambular/models/sklearn_base_lss.py
@@ -41,6 +41,7 @@
     PoissonDistribution,
     Quantile,
     StudentTDistribution,
+    JohnsonSuDistribution
 )
 
 
@@ -48,6 +49,7 @@ class SklearnBaseLSS(BaseEstimator):
     def __init__(self, model, config, **kwargs):
         self.preprocessor_arg_names = [
             "n_bins",
+            "feature_preprocessing",
             "numerical_preprocessing",
             "categorical_preprocessing",
             "use_decision_tree_bins",
@@ -373,6 +375,7 @@ def fit(
             "inversegamma": InverseGammaDistribution,
             "categorical": CategoricalDistribution,
             "quantile": Quantile,
+            "johnsonsu": JohnsonSuDistribution,
         }
 
         if distributional_kwargs is None:
diff --git a/mambular/models/sklearn_base_regressor.py b/mambular/models/sklearn_base_regressor.py
index 94e9bac6..258e084f 100644
--- a/mambular/models/sklearn_base_regressor.py
+++ b/mambular/models/sklearn_base_regressor.py
@@ -25,6 +25,7 @@ class SklearnBaseRegressor(BaseEstimator):
     def __init__(self, model, config, **kwargs):
         self.preprocessor_arg_names = [
             "n_bins",
+            "feature_preprocessing",
             "numerical_preprocessing",
             "categorical_preprocessing",
             "use_decision_tree_bins",