diff --git a/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb b/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
index af760de939525527b65357d040a75a6fa2360b46..cc7d8d3e31f5c560a7b701f59c7298876c821ea0 100644
--- a/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
+++ b/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
@@ -603,7 +603,8 @@
     "\n",
     "        step_timer.done_step(f'Saving data time.')\n",
     "if empty_seq == sum([len(i) for i in mapped_files.values()]):\n",
-    "    raise ValueError(\"No valid trains for RAW data to correct.\")"
+    "    warning(\"No valid trains for RAW data to correct.\")\n",
+    "    sys.exit(0)"
    ]
   },
   {
@@ -631,7 +632,6 @@
    "source": [
     "if skip_plots:\n",
     "    print('Skipping plots')\n",
-    "    import sys\n",
     "    sys.exit(0)"
    ]
   },