diff --git a/ch05/01_main-chapter-code/ch05.ipynb b/ch05/01_main-chapter-code/ch05.ipynb
index 1ae9309..bb17cf5 100644
--- a/ch05/01_main-chapter-code/ch05.ipynb
+++ b/ch05/01_main-chapter-code/ch05.ipynb
@@ -276,7 +276,7 @@
    "source": [
     "- Suppose we have an `inputs` tensor containing the token IDs for 2 training examples (rows)\n",
     "- Corresponding to the `inputs`, the `targets` contain the desired token IDs that we want the model to generate\n",
-    "- Notice that the `targets` are the `inputs` shifted by 1 position, as explained in Chapter 1 when we implemented the data loader"
+    "- Notice that the `targets` are the `inputs` shifted by 1 position, as explained in chapter 2 when we implemented the data loader"
    ]
   },
   {