diff --git a/lectures/makemore/makemore_part4_backprop.ipynb b/lectures/makemore/makemore_part4_backprop.ipynb
index 11b800a8..6e721113 100644
--- a/lectures/makemore/makemore_part4_backprop.ipynb
+++ b/lectures/makemore/makemore_part4_backprop.ipynb
@@ -144,7 +144,8 @@
     "  ex = torch.all(dt == t.grad).item()\n",
     "  app = torch.allclose(dt, t.grad)\n",
     "  maxdiff = (dt - t.grad).abs().max().item()\n",
-    "  print(f'{s:15s} | exact: {str(ex):5s} | approximate: {str(app):5s} | maxdiff: {maxdiff}')"
+    "  shape_ok = dt.shape == t.shape\n",
+    "  print(f'{s:15s} | exact: {str(ex):5s} | approximate: {str(app):5s} | maxdiff: {maxdiff} | shape is correct: {shape_ok}')"
    ]
   },
   {