diff --git a/src/trainers/quantile_trainer.py b/src/trainers/quantile_trainer.py
index c7f6da7..fda821c 100644
--- a/src/trainers/quantile_trainer.py
+++ b/src/trainers/quantile_trainer.py
@@ -675,7 +675,7 @@ class NonAutoRegressiveQuantileRegression(Trainer):
                     title="Final Training Plot",
                     series=f"Sample {actual_idx}",
                     iteration=epoch,
-                    image_path=f"sample_{actual_idx}_plot.png",
+                    local_path=f"sample_{actual_idx}_plot.png",
                 )
 
                 fig2.savefig(f"sample_{actual_idx}_samples_plot.png")
@@ -683,7 +683,7 @@ class NonAutoRegressiveQuantileRegression(Trainer):
                     title="Final Training Samples Plot",
                     series=f"Sample {actual_idx} samples",
                     iteration=epoch,
-                    image_path=f"sample_{actual_idx}_samples_plot.png",
+                    local_path=f"sample_{actual_idx}_samples_plot.png",
                 )
 
             plt.close()
diff --git a/src/trainers/trainer.py b/src/trainers/trainer.py
index a654df1..fb28a9c 100644
--- a/src/trainers/trainer.py
+++ b/src/trainers/trainer.py
@@ -367,7 +367,7 @@ class Trainer:
                     title="Final Training Plot",
                     series=f"Sample {actual_idx}",
                     iteration=epoch,
-                    image_path=f"sample_{actual_idx}_plot.png",
+                    local_path=f"sample_{actual_idx}_plot.png",
                 )
 
                 fig2.savefig(f"sample_{actual_idx}_samples_plot.png")
@@ -375,7 +375,7 @@ class Trainer:
                     title="Final Training Samples Plot",
                     series=f"Sample {actual_idx} samples",
                     iteration=epoch,
-                    image_path=f"sample_{actual_idx}_samples_plot.png",
+                    local_path=f"sample_{actual_idx}_samples_plot.png",
                 )
             plt.close()
 
diff --git a/src/training_scripts/non_autoregressive_quantiles.py b/src/training_scripts/non_autoregressive_quantiles.py
index 0722e17..703146b 100644
--- a/src/training_scripts/non_autoregressive_quantiles.py
+++ b/src/training_scripts/non_autoregressive_quantiles.py
@@ -2,7 +2,9 @@ from src.utils.clearml import ClearMLHelper
 
 #### ClearML ####
 clearml_helper = ClearMLHelper(project_name="Thesis/NAQR: Linear")
-task = clearml_helper.get_task(task_name="NAQR: Linear + Load + PV + Wind + Net Position")
+task = clearml_helper.get_task(
+    task_name="NAQR: Linear + Load + PV + Wind + Net Position"
+)
 task.execute_remotely(queue_name="default", exit_process=True)
 
 from src.policies.PolicyEvaluator import PolicyEvaluator