diff --git a/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb b/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
index 25abd5e1908a5c8069bfeb296615097a31f1d595..bad008604e7ef2521500e98ed63d14d40fd61c53 100644
--- a/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
+++ b/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
@@ -203,7 +203,8 @@
     "    \"blc_set_min\",\n",
     "    \"force_hg_if_below\",\n",
     "    \"force_mg_if_below\",\n",
-    "    \"low_medium_gap\"\n",
+    "    \"low_medium_gap\",\n",
+    "    \"melt_snow\"\n",
     "]"
    ]
   },
@@ -413,7 +414,8 @@
     "    max_pulses,\n",
     "    h5_data_path=h5path,\n",
     "    h5_index_path=h5path_idx,\n",
-    "    corr_bools=corr_bools\n",
+    "    corr_bools=corr_bools,\n",
+    "    gain_mode=gain_mode\n",
     ")\n",
     "\n",
     "agipd_corr.baseline_corr_noise_threshold = -blc_noise_threshold\n",