diff --git a/assignment_0_3_correspondences_template/imagefiltering.ipynb b/assignment_0_3_correspondences_template/imagefiltering.ipynb
index 78e1aad98e75c2d5201e7ece275e89b5e5156351..772243d0eaffbe9c84098baca0a620eb7ea569eb 100644
--- a/assignment_0_3_correspondences_template/imagefiltering.ipynb
+++ b/assignment_0_3_correspondences_template/imagefiltering.ipynb
@@ -550,7 +550,7 @@
     "print (f'A = {A}')\n",
     "patches = extract_antializased_affine_patches(timg1,\n",
     "                                 A, \n",
-    "                                 torch.zeros(A.size(0)).long(),\n",
+    "                                 torch.zeros(A.size(0)).long().reshape(-1,1),\n",
     "                                 32, 1.0)\n",
     "\n",
     "print (f'patches.shape = {patches.shape}')\n",
@@ -591,7 +591,7 @@
     "print (f'A = {A}')\n",
     "patches = extract_antializased_affine_patches(timg1,\n",
     "                                 A, \n",
-    "                                 torch.zeros(A.size(0)).long(),\n",
+    "                                 torch.zeros(A.size(0)).long().reshape(-1,1),\n",
     "                                 32, 1.0)\n",
     "\n",
     "print (f'patches.shape = {patches.shape}')\n",