diff --git a/examples/python/CuTeDSL/notebooks/elementwise_add.ipynb b/examples/python/CuTeDSL/notebooks/elementwise_add.ipynb
index 9cebc273..b4dc8616 100644
--- a/examples/python/CuTeDSL/notebooks/elementwise_add.ipynb
+++ b/examples/python/CuTeDSL/notebooks/elementwise_add.ipynb
@@ -510,7 +510,7 @@
     "             Thread Block\n",
     "                 Tile\n",
     "\n",
-    "Sliced to Thread-Block local sub-tensor (a (16, 128) tile):  gA[((None, None), bidx)]\n",
+    "Sliced to Thread-Block local sub-tensor (a (16, 256) tile):  gA[((None, None), bidx)]\n",
     "\n",
     "    (16,256)   :  (2048,1)\n",
     "     ~~~~~~        ~~~~~~\n",