diff --git a/notebooks/REMI/REMI_Digitize_and_Transform.ipynb b/notebooks/REMI/REMI_Digitize_and_Transform.ipynb
index 3805f619909df6150f232fd4c46f80765e225f67..b9328aafa1e73bc2ad97f9c66699915f090ba90a 100644
--- a/notebooks/REMI/REMI_Digitize_and_Transform.ipynb
+++ b/notebooks/REMI/REMI_Digitize_and_Transform.ipynb
@@ -484,7 +484,11 @@
     "        col = (signal_idx % 2) if signal_idx < 6 else np.s_[:]\n",
     "        ax = fig.add_subplot(grid[row, col])\n",
     "        \n",
-    "        pulse_idx = np.where(np.isfinite(edges[:, signal_idx, 0]))[0][0]\n",
+    "        finite_edges = np.isfinite(edges[:, signal_idx, 0])\n",
+    "        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",
     "        trigger = triggers[pulse_idx]\n",
     "        \n",