diff --git a/home-assignments/HA1/HA1.ipynb b/home-assignments/HA1/HA1.ipynb
index bc051db..a357dc0 100644
--- a/home-assignments/HA1/HA1.ipynb
+++ b/home-assignments/HA1/HA1.ipynb
@@ -910,7 +910,7 @@
     "            + \"use the `ToTensor` transformation to convert the images to tensors.\"\n",
     "        )\n",
     "\n",
-    "    # The imshow commands expects a `numpy array` with shape (3, width, height)\n",
+    "    # The imshow commands expects a `numpy array` with shape (height, width, 3)\n",
     "    # We rearrange the dimensions with `permute` and then convert it to `numpy`\n",
     "    image_data = image_tensor.permute(1, 2, 0).numpy()\n",
     "    height, width, _ = image_data.shape\n",