diff --git a/docs/python/examples.ipynb b/docs/python/examples.ipynb
index 04e613ac1..1d45a3175 100644
--- a/docs/python/examples.ipynb
+++ b/docs/python/examples.ipynb
@@ -41,8 +41,7 @@
     "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",
     "import pandas as pd\n",
-    "import ribasim\n",
-    "from ribasim import Model, Node\n",
+    "from ribasim import Allocation, Model, Node\n",
     "from ribasim.nodes import (\n",
     "    basin,\n",
     "    discrete_control,\n",
@@ -1238,7 +1237,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "model.allocation = ribasim.Allocation(use_allocation=True, timestep=86400)"
+    "model.allocation = Allocation(use_allocation=True, timestep=86400)"
    ]
   },
   {
@@ -1413,7 +1412,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "model = ribasim.Model(\n",
+    "model = Model(\n",
     "    starttime=\"2020-01-01\",\n",
     "    endtime=\"2020-02-01\",\n",
     ")"
@@ -1541,7 +1540,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "model.allocation = ribasim.Allocation(use_allocation=True, timestep=1e5)"
+    "model.allocation = Allocation(use_allocation=True, timestep=1e5)"
    ]
   },
   {
@@ -1697,7 +1696,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.8"
+   "version": "3.12.2"
   }
  },
  "nbformat": 4,