diff --git a/docs/notebooks/Unet.ipynb b/docs/notebooks/Unet2D.ipynb
similarity index 99%
rename from docs/notebooks/Unet.ipynb
rename to docs/notebooks/Unet2D.ipynb
index 57882ab556df34ef033e0d5ef8640fdb5e222492..2fee1dcce21e36172b57d02f8300fc145fbb484b 100644
--- a/docs/notebooks/Unet.ipynb
+++ b/docs/notebooks/Unet2D.ipynb
@@ -97,7 +97,7 @@
    "outputs": [],
    "source": [
     "# defining model\n",
-    "my_model = qim3d.ml.models.UNet(size = 'medium', dropout = 0.25)\n",
+    "my_model = qim3d.ml.models.UNet2D(size = 'medium', dropout = 0.25)\n",
     "# defining augmentation\n",
     "my_aug = qim3d.ml.Augmentation(resize = 'crop', transform_train = 'light')"
    ]
@@ -340,7 +340,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
+   "display_name": "qim3d",
    "language": "python",
    "name": "python3"
   },
@@ -354,7 +354,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.11"
+   "version": "3.10.14"
   }
  },
  "nbformat": 4,