diff --git a/ML/Pytorch/object_detection/YOLO/data/get_data b/ML/Pytorch/object_detection/YOLO/data/get_data
index c7368b10..4970f6fb 100755
--- a/ML/Pytorch/object_detection/YOLO/data/get_data
+++ b/ML/Pytorch/object_detection/YOLO/data/get_data
@@ -7,11 +7,11 @@
                                                               
 ## OR DOWNLOAD FROM HERE (FASTER DOWNLOAD)                                          
 # VOC2007 DATASET                                                              
-wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.ta
-wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar # 
+wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar
+wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar
 
 # VOC2012 DATASET                                                              
-wget http://host.robots.ox.ac.uk/pascal/VOC/voc2012/VOCtrainval_11-May-2012.ta
+wget http://host.robots.ox.ac.uk/pascal/VOC/voc2012/VOCtrainval_11-May-2012.tar
 
 # Extract tar files
 tar xf VOCtrainval_11-May-2012.tar
diff --git a/ML/Pytorch/object_detection/YOLO/train.py b/ML/Pytorch/object_detection/YOLO/train.py
index 6908e8f4..19f27c42 100755
--- a/ML/Pytorch/object_detection/YOLO/train.py
+++ b/ML/Pytorch/object_detection/YOLO/train.py
@@ -36,8 +36,8 @@
 PIN_MEMORY = True
 LOAD_MODEL = False
 LOAD_MODEL_FILE = "overfit.pth.tar"
-IMG_DIR = "data/images"
-LABEL_DIR = "data/labels"
+IMG_DIR = "data/data/images"
+LABEL_DIR = "data/data/labels"
 
 
 class Compose(object):
@@ -84,7 +84,7 @@ def main():
         load_checkpoint(torch.load(LOAD_MODEL_FILE), model, optimizer)
 
     train_dataset = VOCDataset(
-        "data/100examples.csv",
+        "data/train.csv",
         transform=transform,
         img_dir=IMG_DIR,
         label_dir=LABEL_DIR,