diff --git a/docs/notebooks/paper_experiments/paper_2_supervised.ipynb b/docs/notebooks/paper_experiments/paper_2_supervised.ipynb
index 6910de3c..38842fde 100644
--- a/docs/notebooks/paper_experiments/paper_2_supervised.ipynb
+++ b/docs/notebooks/paper_experiments/paper_2_supervised.ipynb
@@ -424,7 +424,7 @@
     "# visble colormap\n",
     "norm = mlp.colors.Normalize( vmin=-1, vmax=1)\n",
     "sm = plt.cm.ScalarMappable(norm=norm,cmap='fessa')\n",
-    "cbar = plt.colorbar(mappable=sm, fraction=0.050, pad=0.02, format='%.1f', ticks=[-1.0, -0.5, 0.0, 0.5, 1.0])\n",
+    "cbar = plt.colorbar(mappable=sm, fraction=0.050, pad=0.02, format='%.1f', ticks=[-1.0, -0.5, 0.0, 0.5, 1.0], ax = ax)\n",
     "cbar.set_label('DeepTDA CV',fontsize=10)\n",
     "\n",
     "# labels\n",
@@ -478,7 +478,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.10"
+   "version": "3.10.8"
   },
   "orig_nbformat": 4,
   "vscode": {