diff --git a/Image-Classification-in-PyTorch/image_classification_using_transfer_learning_in_pytorch.ipynb b/Image-Classification-in-PyTorch/image_classification_using_transfer_learning_in_pytorch.ipynb
index d66db45b4..9346b8320 100644
--- a/Image-Classification-in-PyTorch/image_classification_using_transfer_learning_in_pytorch.ipynb
+++ b/Image-Classification-in-PyTorch/image_classification_using_transfer_learning_in_pytorch.ipynb
@@ -83,7 +83,7 @@
     "bs = 32\n",
     "\n",
     "# Number of classes\n",
-    "num_classes = len(os.listdir(valid_directory))-1  #10#2#257\n",
+    "num_classes = len(os.listdir(valid_directory))  #10#2#257\n",
     "print(num_classes)\n",
     "\n",
     "# Load Data from folders\n",