diff --git a/benchmarks/imaging_ecal/scripts/draw_cluster.py b/benchmarks/imaging_ecal/scripts/draw_cluster.py
index 35dbf0c97deeb0490540edf613f1ac6d019cd8f9..2e72941d3df7747857ee8a92c87311fc87b9843d 100644
--- a/benchmarks/imaging_ecal/scripts/draw_cluster.py
+++ b/benchmarks/imaging_ecal/scripts/draw_cluster.py
@@ -151,7 +151,7 @@ if __name__ == '__main__':
     dfallmcp = get_all_mcp(args.file, args.iev, 'mcparticles2')
     # Select decaying particles
     dftemp = dfallmcp[dfallmcp['g4Parent'] == 1.0]
-    if len(dfdecaymcp) > 0:
+    if len(dftemp) > 0:
         dfdecaymcp = dftemp.copy()
         pdgbase = ROOT.TDatabasePDG()
         for iptl in [0, len(dfdecaymcp) - 1]:
@@ -238,7 +238,8 @@ if __name__ == '__main__':
                           bins=(np.arange(*eta_rg, step=args.topo_size/1000.), np.arange(*phi_rg, step=args.topo_size)),
                           cmap=cmap, cmin=0., pc_kw=dict(alpha=0.8, edgecolor='k'))
     # draw true decaying particle position 
-    ax.scatter(dfdecaymcp['eta'].values, dfdecaymcp['phi'].values, marker='x', color='red', s=22**2, linewidth=5.0)
+    if len(dfdecaymcp) > 0:
+        ax.scatter(dfdecaymcp['eta'].values, dfdecaymcp['phi'].values, marker='x', color='red', s=22**2, linewidth=5.0)
 
     ax.set_ylabel(r'$\phi$ (mrad)', fontsize=32)
     ax.set_xlabel(r'$\eta$', fontsize=32)