diff --git a/benchmarks/imaging_shower_ML/scripts/epcut_scan.py b/benchmarks/imaging_shower_ML/scripts/epcut_scan.py
index 0c35f4a247556a18fb65c07a39e915c24cd2618e..30dddbe42155afebb81fbedc3b231045734e3c45 100644
--- a/benchmarks/imaging_shower_ML/scripts/epcut_scan.py
+++ b/benchmarks/imaging_shower_ML/scripts/epcut_scan.py
@@ -231,7 +231,7 @@ if __name__ == '__main__':
                 # nsamples=npar, # this is redundant
                 nsamples_after_cut=ncut,
                 )
-            ax.hist(ep_centers, weights=vals, bins=50, label='${}$'.format(par.latex),
+            ax.hist(ep_centers, weights=vals/npar, bins=50, label='${}$'.format(par.latex),
                     color=colors[k], ec=colors[k], alpha=0.5)
             texts.append(r'$\epsilon_{{{}}}={:.2f}$%'.format(par.latex, eff*100.))
         cut_data.append(cut_result)
@@ -239,7 +239,7 @@ if __name__ == '__main__':
         ax.legend(fontsize=20, ncol=2, loc='upper center', bbox_to_anchor=(0.5, 1.12),)
         ax.tick_params(labelsize=20)
         ax.set_xlabel('$E/p$', fontsize=20)
-        ax.set_ylabel('Counts', fontsize=20)
+        ax.set_ylabel('Normalized Counts', fontsize=20)
         ax.axvline(x=ep_cut, color='k', ls='--', lw=2)
         ax.text(0.5, 0.97, '\n'.join(texts),
                 transform=ax.transAxes, fontsize=20, va='top', ha='center', bbox=box_props)
diff --git a/benchmarks/imaging_shower_ML/scripts/utils.py b/benchmarks/imaging_shower_ML/scripts/utils.py
index f3cd7538f612447e88afa82a5d016bd6a46cb992..bd3e95363e110e991728a0b47015b0ec91af0ae1 100644
--- a/benchmarks/imaging_shower_ML/scripts/utils.py
+++ b/benchmarks/imaging_shower_ML/scripts/utils.py
@@ -20,7 +20,7 @@ class dotdict(dict):
 # basic information share among all scripts
 imcal_info = dotdict(
         nlayers_img=6,
-        nlayers_scfi=20,
+        nlayers_scfi=16,
         ml_data_key='/ml_data',
         truth_data_key='/truth',
     )