diff --git a/code/compute_results.py b/code/compute_results.py
index 45b5ca7bb912feb57f24f023335cb24d7a70019b..e9c93e18f5d804a39ad98c47893850173ef17407 100644
--- a/code/compute_results.py
+++ b/code/compute_results.py
@@ -188,7 +188,7 @@ if __name__ == "__main__":
             title='Loss values of {} using the best hyperparams'.format(args.dataset_name)
         )
         Plotter.plot_losses(
-            file_path=output_path + 'stage1' + os.sep + 'losses_wo_params.png',
+            file_path=output_path + os.sep + 'losses_wo_params.png',
             all_experiment_scores=[base_wo_params_train_scores, base_wo_params_dev_scores, base_wo_params_test_scores,
                 random_wo_params_train_scores, random_wo_params_dev_scores, random_wo_params_test_scores,
                 omp_wo_params_train_scores, omp_wo_params_dev_scores, omp_wo_params_test_scores],