diff --git a/src/bayesvalidrox/surrogate_models/engine.py b/src/bayesvalidrox/surrogate_models/engine.py
index bcaae6c5756ddc66730b277fc914704dd5eeec8e..e6048900ed28ead0f0fffe0d1a83fd9df05474a0 100644
--- a/src/bayesvalidrox/surrogate_models/engine.py
+++ b/src/bayesvalidrox/surrogate_models/engine.py
@@ -189,12 +189,16 @@ class Engine():
         # Prepare X samples 
         # For training the surrogate use ExpDesign.X_tr, ExpDesign.X is for the model to run on 
         ExpDesign.generate_ED(ExpDesign.n_init_samples,
+<<<<<<< HEAD
 <<<<<<< HEAD
                                               transform=True,
                                               max_pce_deg=np.max(MetaModel.pce_deg))
         
 =======
                               #transform=True,
+=======
+                              # transform=True,
+>>>>>>> f8175f33 (Bug fix: ExpDesign.generate_ED no longer needs 'transform')
                               max_pce_deg=np.max(MetaModel.pce_deg))
 
 >>>>>>> 2306e76d (Decoupled MCMC from BayesInference and improved performance)