diff --git a/ch07/01_main-chapter-code/ch07.ipynb b/ch07/01_main-chapter-code/ch07.ipynb
index 22c031c..0b2129d 100644
--- a/ch07/01_main-chapter-code/ch07.ipynb
+++ b/ch07/01_main-chapter-code/ch07.ipynb
@@ -1634,7 +1634,7 @@
     ")\n",
     "# Alternatively:\n",
     "# from llms_from_scratch.ch05 import (\n",
-    "#    generate_text_simple,\n",
+    "#    generate,\n",
     "#    text_to_token_ids,\n",
     "#    token_ids_to_text\n",
     "# )\n",