diff --git a/notebooks/ePix100/Correction_ePix100_NBC.ipynb b/notebooks/ePix100/Correction_ePix100_NBC.ipynb
index b7db6722739c7d81852329e296cacf48005a8221..bc3a719f39ae39c8a6e2a1228cb51ab48740d75c 100644
--- a/notebooks/ePix100/Correction_ePix100_NBC.ipynb
+++ b/notebooks/ePix100/Correction_ePix100_NBC.ipynb
@@ -636,7 +636,10 @@
     "        \n",
     "        if np.isin('data.pulseId', list(seq_dc[instrument_src].keys())): # some runs are missing 'data.pulseId'\n",
     "            outp_source.create_key(\n",
-    "                \"data.pulseId\", data=list(seq_dc[instrument_src]['data.pulseId'].ndarray().squeeze()), chunks=min(50, len(seq_dc.train_ids)))\n",
+    "                \"data.pulseId\",\n",
+    "                data=list(seq_dc[instrument_src]['data.pulseId'].ndarray()[:, 0]),\n",
+    "                chunks=min(50, len(seq_dc.train_ids)),\n",
+    "            )\n",
     "        \n",
     "        if pattern_classification:\n",
     "            # Add main corrected `data.image.pixels` dataset and store corrected data.\n",