diff --git a/benchmarks/tracking/scripts/tracking_performance.py b/benchmarks/tracking/scripts/tracking_performance.py
index 7a2cb76af88623aa5e8b588c3cf8ae1ca53ed6e8..b776097a9723c58b234b09f02b987cdde67097b1 100644
--- a/benchmarks/tracking/scripts/tracking_performance.py
+++ b/benchmarks/tracking/scripts/tracking_performance.py
@@ -116,6 +116,8 @@ if __name__ == '__main__':
             help='Collection name of MC particles truth info.')
     parser.add_argument('--tracking-collection', dest='coll', default='outputTrackParameters',
             help='Collection name of clusters to plot')
+    parser.add_argument('--tracking-collection-columns', dest='cols', default=['theta', 'phi', 'qOverP'],
+            help='Collection columes of clusters to load')
     args = parser.parse_args()
 
     # multi-threading for RDataFrame
@@ -134,7 +136,7 @@ if __name__ == '__main__':
 
     rdf_rec = ROOT.RDataFrame('events', args.rec_file)
     dfm = thrown_particles_figure(rdf_rec, save=os.path.join(args.outdir, '{}_thrown_particles.png'.format(args.nametag)), mcbranch=args.mc)
-    df = flatten_collection(rdf_rec, args.coll)
+    df = flatten_collection(rdf_rec, args.coll, args.cols)
     df.rename(columns={c: c.replace(args.coll + '.', '') for c in df.columns}, inplace=True)
 
     fig, axs = plt.subplots(3, 2, figsize=(16, 18), dpi=120)