diff --git a/notebooks/Jungfrau/Jungfrau_dark_analysis_all_gains_burst_mode_NBC.ipynb b/notebooks/Jungfrau/Jungfrau_dark_analysis_all_gains_burst_mode_NBC.ipynb
index 3befbb8e77473e2c984218ce7c4915ddf15d8946..a209c075afd73dceef2be5902a096ae7ee382709 100644
--- a/notebooks/Jungfrau/Jungfrau_dark_analysis_all_gains_burst_mode_NBC.ipynb
+++ b/notebooks/Jungfrau/Jungfrau_dark_analysis_all_gains_burst_mode_NBC.ipynb
@@ -312,7 +312,8 @@
     "            # Check if there are wrong bad gain values.\n",
     "            # 1. Exclude empty images.\n",
     "            # 2. Indicate pixels with wrong gain value for any train for each cell.\n",
-    "            gain_avg = np.average(\n",
+    "            # TODO: mean is used to use thresholds for accepting gain values, even if not 0 mean value.\n",
+    "            gain_avg = np.mean(\n",
     "                gain_vals[..., cell_slice_idx][..., nonzero_adc], axis=2)\n",
     "\n",
     "            bad_pixels_map[mod][..., cell_number, gain][gain_avg != raw_g] |= BadPixels.WRONG_GAIN_VALUE.value\n",