diff --git a/benchmarks/dis/analysis/truth_reconstruction.py b/benchmarks/dis/analysis/truth_reconstruction.py
index c1c21b0a61b94b9517d87c0559061fb7c3d1af4d..e726e7ee20660b3903467009e475da9c15c4fdf8 100644
--- a/benchmarks/dis/analysis/truth_reconstruction.py
+++ b/benchmarks/dis/analysis/truth_reconstruction.py
@@ -329,6 +329,7 @@ for i in range(len(MC_list)): #Repeat the following steps for each variable (mom
         ax6.set_title('Photons')
         fig.set_figwidth(20)
         fig.set_figheight(10)
+        x_range = list(ax1.get_xlim())
         ax1.set_title('%s Difference Vs Momentum  %s  %s events\n DETECTOR_CONFIG: %s'%(title_list[i],config,Nevents,Dconfig))
         plt.savefig(os.path.join(r_path, '%s_difference_vs_momentum_%s.png' %  (title_list_n[i],config)))
         plt.close()