From cdaaed546c990bc9eb8b4f8b43b6ab43d5d9f3f7 Mon Sep 17 00:00:00 2001
From: ahmedk <karim.ahmed@xfel.eu>
Date: Fri, 5 Aug 2022 16:21:56 +0200
Subject: [PATCH] small fix after doing last test on reference runs

---
 notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb b/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
index 144d8ecb9..8f91e9263 100644
--- a/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
+++ b/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
@@ -867,7 +867,7 @@
     "display(Markdown((f\"#### The per pixel maximum of train {tid} of the GAIN data\")))\n",
     "\n",
     "fig, ax = plt.subplots(figsize=(18, 10))\n",
-    "gain_max = np.max(gain_train_cells, axis=1)\n",
+    "gain_max = np.max(gain_train_cells, axis=(1, 2))\n",
     "geom.plot_data_fast(\n",
     "    gain_max,\n",
     "    ax=ax,\n",
-- 
GitLab