diff --git a/train.py b/train.py
index cde6b4d..0abcd03 100644
--- a/train.py
+++ b/train.py
@@ -27,7 +27,7 @@ def get_data(size, woof, bs, workers=None):
  #   if   size<=128: path = URLs.IMAGEWOOF_160 if woof else URLs.IMAGENETTE_160
  #   elif size<=224: path = URLs.IMAGEWOOF_320 if woof else URLs.IMAGENETTE_320
    # else          : 
-   if woof:
+    if woof:
         path = URLs.IMAGEWOOF    # if woof 
     else:
         path = URLs.IMAGENETTE