diff --git a/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb b/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
index 3d34b7934b7a7b73899a8fe4ed869dbbdfbfbdf3..62fc6346b585c9400ef16d4b357f5cea1103c702 100644
--- a/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
+++ b/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
@@ -107,11 +107,11 @@
     "    \n",
     "    # Apply PC correction only if requested\n",
     "    if pc_corr:\n",
-    "        corr_bools[\"slopesPC\"] = pc_corr\n",
+    "        corr_bools[\"pc_corr\"] = pc_corr\n",
     "    \n",
     "    # Apply FF correction only if requested\n",
     "    if ff_corr:\n",
-    "        corr_bools[\"slopesFF\"] = ff_corr\n",
+    "        corr_bools[\"ff_corr\"] = ff_corr\n",
     "        \n",
     "    corr_bools[\"adjust_mg_baseline\"] = adjust_mg_baseline\n",
     "    corr_bools[\"do_rel_gain\"] = relative_gain\n",