diff --git a/site/en/tutorials/quickstart/beginner.ipynb b/site/en/tutorials/quickstart/beginner.ipynb
index 6a7811cfde..69e057c915 100644
--- a/site/en/tutorials/quickstart/beginner.ipynb
+++ b/site/en/tutorials/quickstart/beginner.ipynb
@@ -260,7 +260,7 @@
       },
       "outputs": [],
       "source": [
-        "loss_fn(y_train[:1], predictions).numpy()"
+        "loss_fn(y_train[:1], predictions[:1]).numpy()"
       ]
     },
     {