diff --git a/cnn_madmom/segment.py b/cnn_madmom/segment.py
index 1c49adcadfbf2dbcdd34a5f70f4a3f11f69302f1..fdab858d16167748d663eb1cc7c39a2a47bae25a 100644
--- a/cnn_madmom/segment.py
+++ b/cnn_madmom/segment.py
@@ -9,7 +9,7 @@ def segment(songfile):
 
     delay = -4
     smooth = 20
-    threshold = 0.4
+    threshold = 1
 
     cnn = madmom.features.onsets.CNNOnsetProcessor()
     #proc = madmom.features.onsets.OnsetPeakPickingProcessor(threshold=threshold, smooth=smooth, delay=delay, fps=100)
@@ -36,13 +36,15 @@ if __name__ == "__main__":
     activation_function = cnn(songfile, num_channels=1)
     #onsets = proc(activation_function)
 
-    onsets = madmom.features.onsets.peak_picking(activation_function, 0.4, smooth=20)
+    onsets = madmom.features.onsets.peak_picking(activation_function, threshold=1, smooth=20)
+    #onset_smoothed = madmom.audio.signal.smooth(activation_function, 20)
 
     print(onsets/100)
 
     fig, axs = plt.subplots(nrows=2, sharex=True)
     axs[0].imshow(log_spec.T, origin='lower', aspect='auto')
     axs[1].plot(activation_function)
+    #axs[1].plot(onset_smoothed, color='red')
     axs[1].vlines(onsets, 0, 1, colors='red')
 
     plt.show()