diff --git a/ch04/01_main-chapter-code/ch04.ipynb b/ch04/01_main-chapter-code/ch04.ipynb
index f63676e..75e1d23 100644
--- a/ch04/01_main-chapter-code/ch04.ipynb
+++ b/ch04/01_main-chapter-code/ch04.ipynb
@@ -301,10 +301,10 @@
     "torch.manual_seed(123)\n",
     "\n",
     "# create 2 training examples with 5 dimensions (features) each\n",
-    "batch_input = torch.randn(2, 5) \n",
+    "batch_example = torch.randn(2, 5) \n",
     "\n",
     "layer = nn.Sequential(nn.Linear(5, 6), nn.ReLU())\n",
-    "out = layer(batch_input)\n",
+    "out = layer(batch_example)\n",
     "print(out)"
    ]
   },
@@ -492,7 +492,7 @@
    "outputs": [],
    "source": [
     "ln = LayerNorm(emb_dim=5)\n",
-    "out_ln = ln(batch_input)"
+    "out_ln = ln(batch_example)"
    ]
   },
   {