diff --git a/ch04/01_main-chapter-code/ch04.ipynb b/ch04/01_main-chapter-code/ch04.ipynb
index 696ee2b..61f1d74 100644
--- a/ch04/01_main-chapter-code/ch04.ipynb
+++ b/ch04/01_main-chapter-code/ch04.ipynb
@@ -118,6 +118,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "import torch\n",
     "import torch.nn as nn\n",
     "\n",
     "\n",
@@ -195,7 +196,6 @@
    ],
    "source": [
     "import tiktoken\n",
-    "import torch\n",
     "\n",
     "tokenizer = tiktoken.get_encoding(\"gpt2\")\n",
     "\n",