diff --git a/notebooks/AGIPD/AGIPD_Retrieve_Constants_Precorrection.ipynb b/notebooks/AGIPD/AGIPD_Retrieve_Constants_Precorrection.ipynb
index 78f0d185a20bafd3f71b59520962428e258465f4..dbcfe6158dd2c725c11fac2402d2acde19aab711 100644
--- a/notebooks/AGIPD/AGIPD_Retrieve_Constants_Precorrection.ipynb
+++ b/notebooks/AGIPD/AGIPD_Retrieve_Constants_Precorrection.ipynb
@@ -164,7 +164,7 @@
     "\n",
     "# Evaluate gain mode (operation mode)\n",
     "if gain_mode < 0:\n",
-    "    gain_mode = get_gain_mode(control_fn, h5path_ctrl)\n",
+    "    gain_mode = agipdlib.get_gain_mode(control_fn, h5path_ctrl)\n",
     "else:\n",
     "    gain_mode = AgipdGainMode(gain_mode)\n",
     "\n",