diff --git a/CM20315_Loss.ipynb b/CM20315_Loss.ipynb
index 306abc7..8a61aab 100644
--- a/CM20315_Loss.ipynb
+++ b/CM20315_Loss.ipynb
@@ -5,7 +5,7 @@
     "colab": {
       "provenance": [],
       "collapsed_sections": [],
-      "authorship_tag": "ABX9TyMrfCCH1ll/mV1nl3jPkkXz",
+      "authorship_tag": "ABX9TyNWvWC97VuIGwu4VTE1XbO6",
       "include_colab_link": true
     },
     "kernelspec": {
@@ -307,7 +307,7 @@
         "# Return the negative log likelihood of the data under the model\n",
         "def compute_negative_log_likelihood(y_train, mu, sigma):\n",
         "  # TODO -- compute the likelihood of the data -- don't use the likelihood function above -- compute the negative sum of the log probabilities\n",
-        "  # Bottom line of equation 5.3 in the notes\n",
+        "  # Bottom line of equation 5.4 in the notes\n",
         "  # You will need np.sum(), np.log()\n",
         "  # Replace the line below\n",
         "  nll = 0\n",