diff --git a/src/PredictionIntervals/Trainer/TrainNN.py b/src/PredictionIntervals/Trainer/TrainNN.py
index fb3663a..49023b9 100644
--- a/src/PredictionIntervals/Trainer/TrainNN.py
+++ b/src/PredictionIntervals/Trainer/TrainNN.py
@@ -60,7 +60,7 @@ def set_data(self, X: np.array, Y: np.array, Xval: np.array, Yval: np.array):
 
     def set_modelName(self, modelName):
         self.modelName = modelName
-        self._set_folder()
+        self.f = self._set_folder()
 
     def _set_folder(self):
         root = get_project_root()