diff --git a/notebooks/Jungfrau/Jungfrau_dark_analysis_all_gains_burst_mode_NBC.ipynb b/notebooks/Jungfrau/Jungfrau_dark_analysis_all_gains_burst_mode_NBC.ipynb
index 93a402ca0885c6a1e1ae0677bc6cad12468a7fa8..5b4a4640028a4bada1d6729196e24d3b47959121 100644
--- a/notebooks/Jungfrau/Jungfrau_dark_analysis_all_gains_burst_mode_NBC.ipynb
+++ b/notebooks/Jungfrau/Jungfrau_dark_analysis_all_gains_burst_mode_NBC.ipynb
@@ -433,7 +433,12 @@
     "            acelltable -= sc_start\n",
     "        # Only for dynamic medium and low gain runs [forceswitchg1, forceswitchg2] in burst mode.\n",
     "\n",
-    "        if gain_mode == 0 and gain > 0 and memory_cells == 16:\n",
+    "        if (\n",
+    "            gain_mode == 0 and  # dynamic gain mode\n",
+    "            gain > 0 and  # Medium and low runs\n",
+    "            memory_cells == 16 and  # Burst mode\n",
+    "            acelltable.shape[0] == 2  # forceswitchg1 and forceswitchg2 acquired with the MDL device.\n",
+    "        ):\n",
     "            # 255 similar to the receiver which uses the 255\n",
     "            # value to indicate a cell without an image.\n",
     "            # image shape for forceswitchg1 and forceswitchg2 = (1024, 512, 2, trains)\n",