diff --git a/Image Classifier Project.ipynb b/Image Classifier Project.ipynb
index eec7a117..ed1df87e 100644
--- a/Image Classifier Project.ipynb	
+++ b/Image Classifier Project.ipynb	
@@ -231,7 +231,7 @@
     "    \n",
     "    # PyTorch tensors assume the color channel is the first dimension\n",
     "    # but matplotlib assumes is the third dimension\n",
-    "    image = image.numpy().transpose((1, 2, 0))\n",
+    "    image = image.transpose((1, 2, 0))\n",
     "    \n",
     "    # Undo preprocessing\n",
     "    mean = np.array([0.485, 0.456, 0.406])\n",