diff --git a/benchmarks/imaging_ecal/scripts/energy_profile.py b/benchmarks/imaging_ecal/scripts/energy_profile.py
index d5b9b8aa21cd59ec2672abb6273f7d00fe6a57ea..fff925ceb8a62f74cd3b83c2d2faed8944aa7053 100644
--- a/benchmarks/imaging_ecal/scripts/energy_profile.py
+++ b/benchmarks/imaging_ecal/scripts/energy_profile.py
@@ -47,7 +47,7 @@ if __name__ == '__main__':
     dfe.loc[:, 'total_edep'] = dfe.groupby(['event', 'cluster'])['edep'].transform('sum')
     # dfe.loc[:, 'efrac'] = dfe['edep']/dfe['total_edep']
     dfe.loc[:, 'efrac'] = dfe['edep']/(args.energy*1000.)
-    dfe = dfe[dfe['cluster'] == 0]
+    # dfe = dfe[dfe['cluster'] == 0]
 
     os.makedirs(args.outdir, exist_ok=True)