diff --git a/hw4_extra.ipynb b/hw4_extra.ipynb
index 3942c7f..dea9360 100644
--- a/hw4_extra.ipynb
+++ b/hw4_extra.ipynb
@@ -209,7 +209,7 @@
     "\n",
     "<p style=\"text-align: center;\">$X' = X \\; W_o$</p>\n",
     "\n",
-    "Your goal in this part is to return $X$ in the `self.forward` call of `AttentionLayer`. For debugging, you may capture the `probs` variable returned by the inner `MultiHeadAttention` module and store it in an attribute such as `self.probs` of the attention layer.\n",
+    "Your goal in this part is to return $X'$ in the `self.forward` call of `AttentionLayer`. For debugging, you may capture the `probs` variable returned by the inner `MultiHeadAttention` module and store it in an attribute such as `self.probs` of the attention layer.\n",
     "\n",
     "Once finished, you may test your layer with the following test cases."
    ]