diff --git a/Gabriel_Hog.ipynb b/Gabriel_Hog.ipynb
index f318f8f..92f8d04 100644
--- a/Gabriel_Hog.ipynb
+++ b/Gabriel_Hog.ipynb
@@ -43,9 +43,9 @@
     "    smooth_img = skimage.filters.gaussian(img, 10)\n",
     "    sobel_img = skimage.filters.sobel(smooth_img)\n",
     "    if sobel_img.shape[0] == 432:\n",
-    "        hog, hog_image = skimage.feature.hog(skimage.transform.rescale(sobel_img[:,:], 1/11), orientations=9, visualize=True)\n",
+    "        hog, hog_image = skimage.feature.hog(skimage.transform.rescale(sobel_img, 1/11), orientations=9, visualize=True)\n",
     "    else:\n",
-    "        hog, hog_image = skimage.feature.hog(skimage.transform.rescale(sobel_img[:,:], 1/10), orientations=9, visualize=True)\n",
+    "        hog, hog_image = skimage.feature.hog(skimage.transform.rescale(sobel_img, 1/10), orientations=9, visualize=True)\n",
     "    dataset[i] = hog"
    ]
   },