diff --git a/notebooks/REMI/REMI_Digitize_and_Transform.ipynb b/notebooks/REMI/REMI_Digitize_and_Transform.ipynb
index b9328aafa1e73bc2ad97f9c66699915f090ba90a..8d7ce5a454e0517b27cedbb4b18600f2ad371f7e 100644
--- a/notebooks/REMI/REMI_Digitize_and_Transform.ipynb
+++ b/notebooks/REMI/REMI_Digitize_and_Transform.ipynb
@@ -488,8 +488,8 @@
     "        if not finite_edges.any():\n",
     "            continue\n",
     "            \n",
-    "        pulse_idx = np.where(finite_edges)[0][0]\n",
-    "        train_idx = np.where(pulse_idx >= pulse_offsets)[0][-1]\n",
+    "        pulse_idx = finite_edges.nonzero()[0][0]\n",
+    "        train_idx = (pulse_idx >= pulse_offsets).nonzero()[0][-1]\n",
     "        trigger = triggers[pulse_idx]\n",
     "        \n",
     "        sourcekey = remi.get_channel_sourcekey(\n",