diff --git a/src/pages/2-model_creation.py b/src/pages/2-model_creation.py
index 43b0e5fae3142a9dddae711701f9d374ae0ae654..61dd41dbfe341149842eb0eefe3666d80f42f171 100644
--- a/src/pages/2-model_creation.py
+++ b/src/pages/2-model_creation.py
@@ -215,7 +215,7 @@ if not spectra.empty and not y.empty:
 
             Reg = type('obj', (object,), {'model_' : Reg_json['model'], 'best_hyperparams_' : Reg_json['best_lwplsr_params'],
                                           'pred_data_' : [pd.json_normalize(Reg_json[i]) for i in pred]})
-   
+            reg_model = Reg.model_
             Reg.CV_results_ = pd.DataFrame()
             Reg.cv_data_ = {'YpredCV' : {}, 'idxCV' : {}}
             # # set indexes to Reg.pred_data (train, test, folds idx)