diff --git a/notebooks/REMI/REMI_Digitize_and_Transform.ipynb b/notebooks/REMI/REMI_Digitize_and_Transform.ipynb
index bd867ae23db81ee0c1e16781caa773e2179509f6..d91584eb09c8b6d34af919edb4c8ed37ecc4f753 100644
--- a/notebooks/REMI/REMI_Digitize_and_Transform.ipynb
+++ b/notebooks/REMI/REMI_Digitize_and_Transform.ipynb
@@ -304,8 +304,7 @@
     "    discr_func, discr_params = remi.get_discriminator([trigger_edge_channel])\n",
     "\n",
     "    edges = np.zeros(1000, dtype=np.float64)\n",
-    "    num_pulses = discr_func(trigger_trace, edges=edges, **discr_params[0])\n",
-    "    edges = edges[:num_pulses]\n",
+    "    edges, _, num_pulses = discr_func(trigger_trace, edges=edges, **discr_params[0])\n",
     "\n",
     "    first_edge = edges[0]\n",
     "    rel_edges = np.round(edges - first_edge)\n",