diff --git a/ch05/01_main-chapter-code/ch05.ipynb b/ch05/01_main-chapter-code/ch05.ipynb
index c1f866e..6b639d9 100644
--- a/ch05/01_main-chapter-code/ch05.ipynb
+++ b/ch05/01_main-chapter-code/ch05.ipynb
@@ -1247,8 +1247,8 @@
     "            model=model, idx=encoded,\n",
     "            max_new_tokens=50, context_size=context_size\n",
     "        )\n",
-    "        decoded_text = token_ids_to_text(token_ids, tokenizer)\n",
-    "        print(decoded_text.replace(\"\\n\", \" \"))  # Compact print format\n",
+    "    decoded_text = token_ids_to_text(token_ids, tokenizer)\n",
+    "    print(decoded_text.replace(\"\\n\", \" \"))  # Compact print format\n",
     "    model.train()"
    ]
   },