diff --git a/ch03/01_main-chapter-code/ch03.ipynb b/ch03/01_main-chapter-code/ch03.ipynb
index 6d39da1..2fcee0d 100644
--- a/ch03/01_main-chapter-code/ch03.ipynb
+++ b/ch03/01_main-chapter-code/ch03.ipynb
@@ -306,7 +306,7 @@
     "attn_weights_2 = torch.softmax(attn_scores_2, dim=0)\n",
     "\n",
     "print(\"Attention weights:\", attn_weights_2)\n",
-    "print(\"Sum:\", attn_weights.sum())"
+    "print(\"Sum:\", attn_weights_2.sum())"
    ]
   },
   {