diff --git a/benchmarks/dis/analysis/truth_reconstruction.py b/benchmarks/dis/analysis/truth_reconstruction.py
index 69f108f8829657b7c53cc438bf878a9ab7317a42..8999b2c7cf3701a4465dd3f0956e559f869bb17a 100644
--- a/benchmarks/dis/analysis/truth_reconstruction.py
+++ b/benchmarks/dis/analysis/truth_reconstruction.py
@@ -71,7 +71,7 @@ RC_list = [ak.Array(momentum_rc[recID][booll]),                     #Momentum
            ak.Array(phi_rc[recID][booll]),                          #Phi
            -np.log(np.tan((ak.Array(theta_rc[recID][booll]))/2))]   #Eta
 title_list = ['Momentum','Theta','Phi','Eta']
-title_list_n = ['1','2','3','4']
+title_list_n = ['1','2','4','3']
 particle_name_n = {'Electrons':'5','Pions':'6','Photons':'7'}
 ### MC Momentum for different particles list
 M_list = [ak.Array(momentum_mc[simID][booll]),
@@ -271,7 +271,7 @@ for i in range(len(MC_list)): #Repeat the following steps for each variable (mom
     axs[1].set_ylabel('%s_rc'%(title_list[i]))
     axs[1].set_title('%s Correlation'%(title_list[i]))
     fig.suptitle('%s  %s events\n DETECTOR_CONFIG: %s'%(config,Nevents,Dconfig))
-    plt.savefig(os.path.join(r_path, '%s_correlation_%s.png' %  (title_list_n[i],config)))
+    plt.savefig(os.path.join(r_path, '%s%s_%s.png' %  (title_list_n[i],args.config.split('_epic_')[1].strip(),config)))
 
 
 ###################################################################################################
diff --git a/benchmarks/dis/env.sh b/benchmarks/dis/env.sh
index 1afe4f052f0ccad16d789299a7f2b2760954c7f5..fdfdd8e66c77588b153bd3e8e3cc27f30cd3ee6d 100644
--- a/benchmarks/dis/env.sh
+++ b/benchmarks/dis/env.sh
@@ -48,7 +48,7 @@ echo "TMP_PATH:               ${TMP_PATH}"
 ## if not too big).
 RESULTS_PATH="results/${BENCHMARK_TAG}/${BEAM_TAG}/minQ2=${MINQ2}"
 mkdir -p ${RESULTS_PATH}
-mkdir -p "${RESULTS_PATH}/truth_reconstruction"
+
 export RESULTS_PATH=`realpath ${RESULTS_PATH}`
 echo "RESULTS_PATH:           ${RESULTS_PATH}"