diff --git a/aiida_simple.ipynb b/aiida_simple.ipynb
index a3d4c8f..315a9b3 100644
--- a/aiida_simple.ipynb
+++ b/aiida_simple.ipynb
@@ -40,7 +40,7 @@
    "outputs": [],
    "source": [
     "from simple_workflow import add_x_and_y as _add_x_and_y\n",
-    "from simple_workflow import add_x_and_y_and_z"
+    "from simple_workflow import add_x_and_y_and_z as _add_x_and_y_and_z"
    ]
   },
   {
@@ -70,7 +70,7 @@
    "outputs": [],
    "source": [
     "add_x_and_y_task = wg.add_task(\n",
-    "    add_x_and_y,\n",
+    "    _add_x_and_y,\n",
     "    name=\"add_x_and_y\",\n",
     "    x=orm.Int(1),\n",
     "    y=orm.Int(2),\n",
@@ -86,7 +86,7 @@
    "outputs": [],
    "source": [
     "add_x_and_y_and_z_task = wg.add_task(\n",
-    "    add_x_and_y_and_z,\n",
+    "    _add_x_and_y_and_z,\n",
     "    name=\"add_x_and_y_and_z\",\n",
     "    x=add_x_and_y_task.outputs.x,\n",
     "    y=add_x_and_y_task.outputs.y,\n",