diff --git a/ch04/01_main-chapter-code/ch04.ipynb b/ch04/01_main-chapter-code/ch04.ipynb
index bc46df1..61f1d74 100644
--- a/ch04/01_main-chapter-code/ch04.ipynb
+++ b/ch04/01_main-chapter-code/ch04.ipynb
@@ -196,7 +196,6 @@
    ],
    "source": [
     "import tiktoken\n",
-    "import torch\n",
     "\n",
     "tokenizer = tiktoken.get_encoding(\"gpt2\")\n",
     "\n",