diff --git a/docs/quickstart.md b/docs/quickstart.md
index 8211b69d69..573f5c0d08 100644
--- a/docs/quickstart.md
+++ b/docs/quickstart.md
@@ -205,11 +205,11 @@ for step in range(1000):
   opt.step()
 
   # calculate accuracy
-  pred = out.argmax(axis=-1).numpy()
+  pred = out.argmax(axis=-1)
   acc = (pred == labels).mean()
 
   if step % 100 == 0:
-    print(f"Step {step+1} | Loss: {loss.numpy()} | Accuracy: {acc}")
+    print(f"Step {step+1} | Loss: {loss.numpy()} | Accuracy: {acc.numpy()}")
 ```
 
 ## Evaluation