diff --git a/lessons/python/functions.ipynb b/lessons/python/functions.ipynb
index bc7014c..f425cd4 100644
--- a/lessons/python/functions.ipynb
+++ b/lessons/python/functions.ipynb
@@ -499,7 +499,7 @@
    "source": [
     "def diffusion_example():\n",
     "    \"\"\"An example of using `solve1d` in a diffusion problem.\"\"\"\n",
-    "    print(example.__doc__)\n",
+    "    print(diffusion_example.__doc__)\n",
     "    D = 100  # diffusivity\n",
     "    Lx = 10  # domain length\n",
     "    dx = 0.5  # grid spacing\n",
@@ -522,7 +522,7 @@
     "write functions to address the smaller pieces,\n",
     "then assemble the functions to solve the problem.\n",
     "\n",
-    "Run the example `example`."
+    "Run the example `diffusion_example`."
    ]
   },
   {