diff --git a/ch06/01_main-chapter-code/ch06.ipynb b/ch06/01_main-chapter-code/ch06.ipynb
index 7203226..ead9e4f 100644
--- a/ch06/01_main-chapter-code/ch06.ipynb
+++ b/ch06/01_main-chapter-code/ch06.ipynb
@@ -1694,7 +1694,7 @@
    "id": "90521a9a-639c-4c7f-a5c0-aca8fa5d4c1b",
    "metadata": {},
    "source": [
-    "- Let's check the initial classification accuracy before we start training the model:"
+    "- Let's check the initial classification accuracy before we start training the model"
    ]
   },
   {
@@ -2158,7 +2158,9 @@
     "    \" selected to receive $1000 cash or a $2000 award.\"\n",
     ")\n",
     "\n",
-    "print(classify_review(text_1, model, tokenizer, device, max_length=train_dataset.max_length))"
+    "print(classify_review(\n",
+    "    text_1, model, tokenizer, device, max_length=train_dataset.max_length\n",
+    "))"
    ]
   },
   {
@@ -2187,7 +2189,9 @@
     "    \" for dinner tonight? Let me know!\"\n",
     ")\n",
     "\n",
-    "print(classify_review(text_2, model, tokenizer, device, max_length=train_dataset.max_length))"
+    "print(classify_review(\n",
+    "    text_2, model, tokenizer, device, max_length=train_dataset.max_length\n",
+    "))"
    ]
   },
   {