diff --git a/notebooks/pnCCD/Correct_pnCCD_NBC.ipynb b/notebooks/pnCCD/Correct_pnCCD_NBC.ipynb
index 124216bbf643a14a7e2de4f9f8674d6d60ba1905..b19904f03da1090873f98f2dcb44e7e2b9c75ed2 100644
--- a/notebooks/pnCCD/Correct_pnCCD_NBC.ipynb
+++ b/notebooks/pnCCD/Correct_pnCCD_NBC.ipynb
@@ -117,6 +117,7 @@
     "\n",
     "from cal_tools.tools import (\n",
     "    calcat_creation_time,\n",
+    "    exit_notebook,\n",
     "    write_constants_fragment,\n",
     ")\n",
     "from cal_tools.step_timing import StepTimer"
@@ -784,8 +785,7 @@
     "    step_timer.done_step(f'Storing data.')\n",
     "# Exit and raise warning if there are no data to correct for all sequences.\n",
     "if empty_seq == len(seq_files):\n",
-    "    warning(\"No valid trains for RAW data to correct.\")\n",
-    "    sys.exit(0)"
+    "    exit_notebook(\"No valid trains for RAW data to correct.\")"
    ]
   },
   {