diff --git a/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb b/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
index 17f443b9756df72851ab97e0b3ffdc2b1eb0c0d4..8b3b4d462706af4f265260fcc4b0d1b6cfe6ceec 100644
--- a/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
+++ b/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
@@ -602,7 +602,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)"
    ]
   },
   {
@@ -630,7 +631,6 @@
    "source": [
     "if skip_plots:\n",
     "    print('Skipping plots')\n",
-    "    import sys\n",
     "    sys.exit(0)"
    ]
   },