diff --git a/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb b/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
index c70815791b0c5c9187f8a2694bef9e309772b494..de2016ff23eb109a5326c7f09bfa7f76683861d1 100644
--- a/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
+++ b/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
@@ -101,6 +101,7 @@
     "from cal_tools.step_timing import StepTimer\n",
     "from cal_tools.tools import (\n",
     "    calcat_creation_time,\n",
+    "    exit_notebook,\n",
     "    map_seq_files,\n",
     "    write_constants_fragment,\n",
     ")\n",
@@ -738,8 +739,7 @@
     "\n",
     "        step_timer.done_step(f'Saving data time.')\n",
     "if empty_seq == sum([len(i) for i in mapped_files.values()]):\n",
-    "    warning(\"No valid trains for RAW data to correct.\")\n",
-    "    sys.exit(0)"
+    "    exit_notebook(\"No valid trains for RAW data to correct.\")"
    ]
   },
   {
@@ -766,8 +766,7 @@
    "outputs": [],
    "source": [
     "if skip_plots:\n",
-    "    print('Skipping plots')\n",
-    "    sys.exit(0)"
+    "    exit_notebook('Skipping plots')"
    ]
   },
   {