diff --git a/benchmarks/imaging_shower_ML/options/imaging_ml_data.py b/benchmarks/imaging_shower_ML/options/imaging_ml_data.py
index 15a9d5b793815a965ab82e30c54b93bc78bf1a2d..2e7fd7e42957e09e10809feeafc9a825eb4c401e 100644
--- a/benchmarks/imaging_shower_ML/options/imaging_ml_data.py
+++ b/benchmarks/imaging_shower_ML/options/imaging_ml_data.py
@@ -27,6 +27,7 @@ if kwargs['nev'] < 1:
 print(kwargs)
 # get sampling fraction from system environment variable, 1.0 by default
 sf = float(os.environ.get('JUGGLER_SAMP_FRAC', '1.0'))
+print(kwargs['compact'])
 
 geo_service  = GeoSvc('GeoSvc', detectors=[f.strip() for f in kwargs['compact'].split(',')])
 podev = EICDataSvc('EventDataSvc', inputs=[f.strip() for f in kwargs['input'].split(',')])
@@ -103,8 +104,8 @@ becal_scfi_reco = CalHitReco('becal_scfi_reco',
         thresholdFactor=5.0,
         readoutClass='EcalBarrelScFiHits',
         layerField='layer',
-        sectorField='module',
-        localDetFields=['system', 'module'], # use local coordinates in each module (stave)
+        sectorField='sector',
+        localDetFields=['system', 'sector'], # use local coordinates in each module (stave)
         **becal_scfi_daq)
 
 # merge hits in different layer (projection to local x-y plane)
diff --git a/benchmarks/imaging_shower_ML/scripts/ml_imaging_data.py b/benchmarks/imaging_shower_ML/scripts/ml_imaging_data.py
index 61eee6238fe4fef26e49224159e1b9e6d11fa74f..72c484ab3b3215f3b67a6232200e319db18f43e5 100644
--- a/benchmarks/imaging_shower_ML/scripts/ml_imaging_data.py
+++ b/benchmarks/imaging_shower_ML/scripts/ml_imaging_data.py
@@ -204,9 +204,9 @@ if __name__ == '__main__':
                 'momentum.x', 'momentum.y', 'momentum.z',
                 'mass'
                 ])
+        dfm.loc[:, 'event'] = mevs
         # select incident particles
         dfm = dfm[dfm['generatorStatus'].isin([0, 1])]
-        dfm.loc[:, 'event'] = mevs
         # NOTE: assumed single particles
         dfm = dfm.groupby('event').first()
         # True momentum