diff --git a/benchmarks/dis/analysis/truth_reconstruction.py b/benchmarks/dis/analysis/truth_reconstruction.py
index 3221cebe4498619804eaca9c28d649a6e0f76067..5f2003d5703887bd13b8d9b1088b0e6f67ff70eb 100644
--- a/benchmarks/dis/analysis/truth_reconstruction.py
+++ b/benchmarks/dis/analysis/truth_reconstruction.py
@@ -173,17 +173,9 @@ for i in range(len(MC_list)): #Repeat the following steps for each variable (mom
 ###################################################################################################
 
     if i > 0: #for each variable theta, phi, and eta
-        for j in range(len(X_list)): #Repeat the following steps for each particle (pions,protons,electrons,neutrons,photons)
-            X = X_list[j]
-            Y = Y_list[j]
-            M_mc = M_list[j]
-            boolean_M = np.ones_like(M_mc) == 1
-            X_s = np.array(ak.flatten(X[boolean_M])) 
-            Y_s = np.array(ak.flatten(Y[boolean_M])) 
-            M_s = np.array(ak.flatten(M_mc))
-            ratio = np.array((ak.Array(Y_s)-(ak.Array(X_s))))
+        for j in range(len(M_list)): #Repeat the following steps for each particle (pions,protons,electrons,neutrons,photons)
+            M_s = np.array(ak.flatten(M_list[j]))
             X_plot[j] = M_s
-            Y_plot[j] = ratio
 
         fig = plt.figure()
         gs = fig.add_gridspec(3, 2, wspace=0)