diff --git a/notebooks/detr_attention.ipynb b/notebooks/detr_attention.ipynb
index 61d248d0b..4c136babb 100644
--- a/notebooks/detr_attention.ipynb
+++ b/notebooks/detr_attention.ipynb
@@ -1302,7 +1302,7 @@
         "fcenter_ax = fig.add_subplot(gs[:, 1:-1])\n",
         "fcenter_ax.imshow(im)\n",
         "for (y, x) in idxs:\n",
-        "    scale = im.height / img.shape[-1]\n",
+        "    scale = im.height / img.shape[-2]\n",
         "    x = ((x // fact) + 0.5) * fact\n",
         "    y = ((y // fact) + 0.5) * fact\n",
         "    fcenter_ax.add_patch(plt.Circle((x * scale, y * scale), fact // 2, color='r'))\n",
@@ -1619,4 +1619,4 @@
       ]
     }
   ]
-}
\ No newline at end of file
+}