diff --git a/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb b/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
index 962d0d6cb514b0edd8a34f8867ad557595b2e879..932269330e883d043a38182ecbac434e7640b845 100644
--- a/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
+++ b/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
@@ -668,11 +668,9 @@
     "\n",
     "            img_counts = pool.map(agipd_corr.apply_selected_pulses, range(len(file_batch)))\n",
     "            step_timer.done_step(\"Applying selected cells after common mode correction\")\n",
-    "\n",
-    "        # Perform image-wise correction\n",
-    "        if any(agipd_corr.pc_bools):\n",
-    "            pool.starmap(agipd_corr.gain_correction, imagewise_chunks(img_counts))\n",
-    "            step_timer.done_step(\"Gain corrections\")\n",
+    "        # Perform image-wise correction\"\n",
+    "        pool.starmap(agipd_corr.gain_correction, imagewise_chunks(img_counts))\n",
+    "        step_timer.done_step(\"Gain corrections\")\n",
     "\n",
     "        # Save corrected data\n",
     "        pool.starmap(agipd_corr.write_file, [\n",