diff --git a/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb b/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
index a247e97ed7675650f86c99352df19d91495d411e..79f35f8a7151201e45b68e73655e1b0832400f1a 100644
--- a/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
+++ b/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
@@ -831,8 +831,9 @@
     "        **mean_plot_kwargs\n",
     "    )\n",
     "else:\n",
-    "    ax.imshow(corrected_mean.squeeze(), aspect=aspect, **mean_plot_kwargs)\n",
-    "    \n",
+    "    corr = ax.imshow(corrected_mean.squeeze(), aspect=aspect, **mean_plot_kwargs)\n",
+    "    plt.colorbar(corr)\n",
+    "\n",
     "ax.set_title(f'{karabo_id} - Mean CORRECTED', size=18)\n",
     "\n",
     "plt.show()"
@@ -858,7 +859,8 @@
     "        **mean_plot_kwargs\n",
     "    )\n",
     "else:\n",
-    "    ax.imshow(corrected_mean.squeeze(), aspect=aspect, **mean_plot_kwargs)\n",
+    "    corr = ax.imshow(corrected_masked_mean.squeeze(), aspect=aspect, **mean_plot_kwargs)\n",
+    "    plt.colorbar(corr)\n",
     "\n",
     "ax.set_title(f'{karabo_id} - Mean CORRECTED with mask', size=18)\n",
     "\n",
@@ -889,7 +891,8 @@
     "        **single_plot_kwargs\n",
     "    )\n",
     "else:\n",
-    "    ax.imshow(corrected_train.squeeze(), aspect=aspect, **single_plot_kwargs)\n",
+    "    corr = ax.imshow(corrected_train.squeeze(), aspect=aspect, **single_plot_kwargs)\n",
+    "    plt.colorbar(corr)\n",
     "\n",
     "ax.set_title(f\"{karabo_id} - CORRECTED train: {tid}\", size=18)\n",
     "\n",
@@ -1059,7 +1062,9 @@
     "        colorbar={'shrink': 1, 'pad': 0.01},\n",
     "    )\n",
     "else:\n",
-    "    ax.imshow(np.log2(mask_train).squeeze(), vmin=0, vmax=32, cmap='jet', aspect=aspect)\n",
+    "    mask = ax.imshow(\n",
+    "        mask_train.squeeze(), vmin=0, vmax=32, cmap='jet', aspect=aspect)\n",
+    "    plt.colorbar(mask)\n",
     "\n",
     "plt.show()"
    ]