diff --git a/notebooks/LPD/LPD_Correct_Fast.ipynb b/notebooks/LPD/LPD_Correct_Fast.ipynb
index 4a93071a535f430c18d321c2991fadb7bf579578..7457d3b09ce8d2f42c5d32c60ef4d8a0691b599e 100644
--- a/notebooks/LPD/LPD_Correct_Fast.ipynb
+++ b/notebooks/LPD/LPD_Correct_Fast.ipynb
@@ -501,17 +501,17 @@
     "            # Replicate corrected cell and pulse IDs from high gain to other gains.\n",
     "            tmp_cell = np.zeros_like(in_cell)\n",
     "            tmp_pulse = np.zeros_like(in_pulse)\n",
-    "            sel = np.zeros_like(in_cell)\n",
+    "            sel = np.zeros_like(in_cell, dtype=np.bool_)\n",
     "\n",
     "            cursor_in = cursor_out = 0\n",
-    "            for afc in actual_frame_count:\n",
+    "            for afc in actual_frame_counts:\n",
     "                for gg in range(3):\n",
     "                    tmp_cell[cursor_out : cursor_out+afc] = in_cell[cursor_in : cursor_in+afc]\n",
     "                    tmp_pulse[cursor_out : cursor_out+afc] = in_pulse[cursor_in : cursor_in+afc]\n",
-    "                    sel[cursor_out : cursor_out+afc] = 1\n",
+    "                    sel[cursor_out : cursor_out+afc] = True\n",
     "                    cursor_out += afc + drop_last_frames_parallelgain\n",
     "\n",
-    "                cursor_in += afc + drop_last_frames_parallelgain\n",
+    "                cursor_in += 3 * (afc + drop_last_frames_parallelgain)\n",
     "                \n",
     "            # Apply the selection to drop trailing frames\n",
     "            in_cell = tmp_cell[sel]\n",