diff --git a/notebooks/pnCCD/Correct_pnCCD_NBC.ipynb b/notebooks/pnCCD/Correct_pnCCD_NBC.ipynb
index 040f004c068f363bfc9bd2b79898482bed473d14..8726a833f553159dd593c3ece23304ba77d798f2 100644
--- a/notebooks/pnCCD/Correct_pnCCD_NBC.ipynb
+++ b/notebooks/pnCCD/Correct_pnCCD_NBC.ipynb
@@ -90,7 +90,6 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import datetime\n",
     "import os\n",
     "import sys\n",
     "import warnings\n",
@@ -607,8 +606,8 @@
     "    \"\"\"\n",
     "    # Create CORR files and add corrected data sections.\n",
     "    image_counts = seq_dc[instrument_src, \"data.image\"].data_counts(labelled=False)\n",
+    "    dataset_chunk = ((chunk_size_idim,) + corr_arrays[\"pixels\"].shape[1:])  # e.g. (1, pixels_x, pixels_y) \n",
     "    with DataFile(out_file, 'w') as ofile:\n",
-    "        dataset_chunk = ((chunk_size_idim,) + corr_arrays[\"pixels\"].shape[1:])  # e.g. (1, pixels_x, pixels_y) \n",
     "\n",
     "        # Create INDEX datasets.\n",
     "        ofile.create_index(seq_dc.train_ids, from_file=seq_dc.files[0])\n",
@@ -662,7 +661,7 @@
     "    img_dc = seq_dc[instrument_src, \"data.image\"]\n",
     "\n",
     "    dshape = seq_dc[instrument_src, \"data.image\"].shape\n",
-    "    seq_dc[instrument_src, \"data.image\"]\n",
+    "\n",
     "    n_trains = dshape[0]\n",
     "    corr_ntrains = dshape[0]  # number of available trains to correct.\n",
     "    all_train_ids = img_dc.train_ids    # All trains including trains with no data.\n",