diff --git a/src/bayesvalidrox/surrogate_models/surrogate_models.py b/src/bayesvalidrox/surrogate_models/surrogate_models.py
index d6c9689440f37df597d9c790088cf13ddb36f199..b94a651e459292e2d11d0a645cfc51fa9411ac71 100644
--- a/src/bayesvalidrox/surrogate_models/surrogate_models.py
+++ b/src/bayesvalidrox/surrogate_models/surrogate_models.py
@@ -90,7 +90,8 @@ def _preprocessing_fit(fit_function):
 
         # Build the input space
         if self.InputSpace is None:
-            self.InputSpace = InputSpace(self.input_obj)
+            print(self.meta_model_type)
+            self.InputSpace = InputSpace(self.input_obj, self.meta_model_type)
             n_init_samples = X.shape[0]
             self.InputSpace.n_init_samples = n_init_samples
             # TODO: _pce_deg not necessarily available, generalize this!
@@ -379,12 +380,13 @@ class MetaModel:
 
     """
 
-    def __init__(self, input_obj, bootstrap_method='fast',
+    def __init__(self, input_obj, meta_model_type = '', bootstrap_method='fast',
                  n_bootstrap_itrs=1, dim_red_method='no', is_gaussian=False,
                  verbose=False):
 
         # Inputs
         self.input_obj = input_obj
+        self.meta_model_type = meta_model_type
         self.bootstrap_method = bootstrap_method
         self.n_bootstrap_itrs = n_bootstrap_itrs
         self.dim_red_method = dim_red_method