diff --git a/pes_to_spec/model.py b/pes_to_spec/model.py
index 41014f6cc5089dbe6a4002801a9df67524fa7a59..af421a4079db09b8c3bbb53599c4a5b5e1daa9ac 100644
--- a/pes_to_spec/model.py
+++ b/pes_to_spec/model.py
@@ -131,7 +131,7 @@ class Model(object):
         """
         # reduce on channel and on train ID
         sum_low_res = - np.mean(sum(list(low_res_data.values())), axis=0)
-        widths = np.arange(10, 50, step=5)
+        widths = np.arange(10, 50, step=1)
         peak_idx = find_peaks_cwt(sum_low_res, widths)
         if len(peak_idx) < 1:
             raise PromptNotFoundError()
@@ -158,7 +158,7 @@ class Model(object):
         ax.set(title="",
                xlabel="Photon Spectrometer channel",
                ylabel="Sum of all Photon Spectrometer channels")
-        plt.axvline(100,
+        plt.axvline(peak_idx,
                 linewidth=3,
                 ls="--",
                 color='r',
diff --git a/scripts/test_analysis.py b/scripts/test_analysis.py
index 56b924ccf76a75107aa92439d344f914e5d62294..3d848dcf03361feeece05f32ee64c11761178af1 100755
--- a/scripts/test_analysis.py
+++ b/scripts/test_analysis.py
@@ -111,7 +111,7 @@ def main():
                  n_pca_hr=20,
                  high_res_sigma=0.2,
                  tof_start=None,
-                 delta_tof=400,
+                 delta_tof=200,
                  validation_size=0.05)
 
     train_idx = np.isin(tids, train_tids)