diff --git a/notebooks/AGIPD/Characterize_AGIPD_Gain_FlatFields_Summary.ipynb b/notebooks/AGIPD/Characterize_AGIPD_Gain_FlatFields_Summary.ipynb
index 9673c4959c0692cebf12319867ff414afb414756..1754a260c6e47b86611b8b1f5d6ec2a759b1df6c 100644
--- a/notebooks/AGIPD/Characterize_AGIPD_Gain_FlatFields_Summary.ipynb
+++ b/notebooks/AGIPD/Characterize_AGIPD_Gain_FlatFields_Summary.ipynb
@@ -591,10 +591,10 @@
     "    run_data = RunDirectory(run_folder, include)\n",
     "    if tid:\n",
     "        tid, data = run_data.select('*/DET/*', source).train_from_id(tid)\n",
-    "        return tid, stack_detector_data(data, source,modules=nmods)\n",
+    "        return tid, stack_detector_data(data, source, modules=nmods)\n",
     "    else:\n",
     "        for tid, data in run_data.select('*/DET/*', source).trains(require_all=True):\n",
-    "            return tid, stack_detector_data(data, source,modules=nmods)\n",
+    "            return tid, stack_detector_data(data, source, modules=nmods)\n",
     "    return None, None\n",
     "\n",
     "\n",
@@ -655,8 +655,6 @@
     "fig = plt.figure(figsize=(20,20))\n",
     "ax = fig.add_subplot(111)\n",
     "cdata = np.nanmean(corrected, axis=0)\n",
-    "#Remove me: the scale for comparision should be the same as for orig data\n",
-    "#vmin, vmax = get_range(cdata, 5)\n",
     "ax = geom.plot_data_fast(cdata, ax=ax, cmap=\"jet\", vmin=vmin, vmax=vmax, figsize=(20,20))\n",
     "_ = ax.set_title(\"Corrected data, mean across one train\")"
    ]