diff --git a/benchmarks/imaging_shower_ML/options/imaging_ml_data.py b/benchmarks/imaging_shower_ML/options/imaging_ml_data.py
index 4b8188c94be9d272276070d9d3ce76ad682c7aa7..fbdcf9328fd7b6c69939df2580d66bed1e9cb8b2 100644
--- a/benchmarks/imaging_shower_ML/options/imaging_ml_data.py
+++ b/benchmarks/imaging_shower_ML/options/imaging_ml_data.py
@@ -12,7 +12,7 @@ from Configurables import Jug__Reco__CalorimeterHitsMerger as CalHitsMerger
 from Configurables import Jug__Reco__ImagingPixelMerger as MLDataMerger
 from Configurables import Jug__Reco__ImagingPixelDataSorter as MLDataSorter
 from Configurables import Jug__Reco__ImagingPixelDataCombiner as MLDataCombiner
-
+from Configurables import Jug__Reco__ImagingPixelDataShaper as MLDataShaper
 
 # input arguments through environment variables
 kwargs = dict()
@@ -123,7 +123,11 @@ becal_combiner = MLDataCombiner('becal_combiner',
         layerIncrement=100,
         rule=kwargs['combine'])
 
-        
+
+# shape layers
+becal_shaper = MLDataShaper('becal_combiner',
+        inputHits=becal_combiner.outputHits)
+
 
 podout.outputCommands = [
 #        'keep *',
@@ -138,7 +142,7 @@ ApplicationMgr(
     TopAlg=[podin,
             becal_img_digi, becal_img_reco, becal_img_merger, becal_img_sorter,
             becal_scfi_digi, becal_scfi_reco, becal_scfi_merger, becal_scfi_sorter,
-            becal_combiner,
+            becal_combiner, becal_shaper,
             podout],
     EvtSel='NONE',
     EvtMax=kwargs['nev'],