diff --git a/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb b/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
index 2048c87e63883b418491347c44e655fff8d04a08..f8e99656a9694b6776c78ea8b74a5ba8daf2f2f8 100644
--- a/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
+++ b/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
@@ -452,7 +452,7 @@
    "outputs": [],
    "source": [
     "if use_litframe_finder != 'off':\n",
-    "    from extra_redu import make_litframe_finder, SourceNotFound as RequiredSourceNotFound\n",
+    "    from extra_redu import make_litframe_finder, LitFrameFinderError\n",
     "    from extra_redu.litfrm.utils import litfrm_run_report\n",
     "\n",
     "    if use_litframe_finder not in ['auto', 'offline', 'online']:\n",
@@ -482,7 +482,7 @@
     "                ).format(frmintf=frmintf, trsintf=trsintf, **rec)\n",
     "            )\n",
     "        cell_sel = LitFrameSelection(r, train_ids, max_pulses, energy_threshold)\n",
-    "    except RequiredSourceNotFound as err:\n",
+    "    except LitFrameFinderError as err:\n",
     "        print(\"Cannot use AgipdLitFrameFinder due to:\")\n",
     "        print(err)\n",
     "        cell_sel = CellRange(max_pulses, max_cells=mem_cells)\n",
diff --git a/setup.py b/setup.py
index 3f98059d3ef06d0f7673dedb4c7b09f12b192a65..aae7823b411bdbf59ab410e4e3b2d4d4fe510821 100644
--- a/setup.py
+++ b/setup.py
@@ -102,8 +102,8 @@ install_requires = [
         "sphinx==1.8.5",
         "tabulate==0.8.6",
         "traitlets==4.3.3",
-	"xarray==2022.3.0",
-        "EXtra-redu==0.0.4",
+        "xarray==2022.3.0",
+        "EXtra-redu==0.0.5",
 ]
 
 if "readthedocs.org" not in sys.executable: