diff --git a/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb b/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
index db092d2dd5836efdb4112d82ea10b977b7ed1caa..eaed1327aa54899886e988f332a6e046bfd1dd14 100644
--- a/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
+++ b/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
@@ -653,7 +653,7 @@
     "\n",
     "print(\"Preparing constants (\"\n",
     "      f\"FF: {agipd_corr.corr_bools.get('xray_corr', False)}, \"\n",
-    "      f\"{rel_gain}: {any(agipd_corr.rel_gain_bools)}, \"\n",
+    "      f\"{rel_gain_alias}: {any(agipd_corr.rel_gain_bools)}, \"\n",
     "      f\"BLC: {any(agipd_corr.blc_bools)})\")\n",
     "# Display retrieved calibration constants timestamps\n",
     "agipd_cal.display_markdown_retrieved_constants(metadata=agipd_metadata)"
@@ -757,7 +757,7 @@
     "\n",
     "    # Store few time stamps if exists\n",
     "    # Add NA to keep array structure\n",
-    "    for key in ['Offset', f'Slopes{rel_gain_mode}', 'SlopesFF']:\n",
+    "    for key in ['Offset', f'Slopes{rel_gain_alias}', 'SlopesFF']:\n",
     "        if key in mod_mdata:\n",
     "            module_timestamps[key] = mod_mdata[key][\"begin_validity_at\"]\n",
     "        else:\n",
diff --git a/notebooks/AGIPD/AGIPD_Correct_and_Verify_Summary_NBC.ipynb b/notebooks/AGIPD/AGIPD_Correct_and_Verify_Summary_NBC.ipynb
index 36e21e9220cc0fcda5f1f4f4a235fcdef1208034..7c3aba0dff2482aae2997f3ad1e3a0ca58503218 100644
--- a/notebooks/AGIPD/AGIPD_Correct_and_Verify_Summary_NBC.ipynb
+++ b/notebooks/AGIPD/AGIPD_Correct_and_Verify_Summary_NBC.ipynb
@@ -18,7 +18,8 @@
     "metadata_folder = \"\"  # Directory containing calibration_metadata.yml when run by xfel-calibrate\n",
     "karabo_id = \"SPB_DET_AGIPD1M-1\" # karabo karabo_id\n",
     "modules = [-1]\n",
-    "karabo_da = ['-1']  # a list of data aggregators names, Default [-1] for selecting all data aggregators"
+    "karabo_da = ['-1']  # a list of data aggregators names, Default [-1] for selecting all data aggregators\n",
+    "rel_gain = \"off\"  # Select relative gain correction. Choices [`PC`, `CS`, `off`]. (`PC`: Pulse Capacitor, `CS`: Current Source, `off`: Disable relative gain correction). Default: off."
    ]
   },
   {
@@ -105,8 +106,8 @@
     "    display(Latex(tabulate.tabulate(table,\n",
     "                 tablefmt=\"latex\",\n",
     "                 headers=[\"Timestamps\", \"Modules and sequences\"])))\n",
-    "\n",
-    "for const in ['Offset', 'SlopesPC', 'SlopesFF']:\n",
+    "rel_gain_alias = \"CS\" if rel_gain.lower() == \"cs\" else \"PC\"\n",
+    "for const in ['Offset', f'Slopes{rel_gain_alias}', 'SlopesFF']:\n",
     "    print_const_table(const)"
    ]
   }