diff --git a/test.py b/test.py
index 014a7d29..431515ba 100644
--- a/test.py
+++ b/test.py
@@ -32,7 +32,7 @@ num_classes = int(data_config['classes'])
 model = Darknet(opt.cfg, opt.img_size)
 
 # Load weights
-weights_path = 'checkpoints/yolov3.pt'
+weights_path = 'checkpoints/yolov3.weights'
 if weights_path.endswith('.weights'):  # darknet format
     load_weights(model, weights_path)
 elif weights_path.endswith('.pt'):  # pytorch format
diff --git a/utils/datasets.py b/utils/datasets.py
index ed71c31b..07647166 100755
--- a/utils/datasets.py
+++ b/utils/datasets.py
@@ -68,7 +68,7 @@ class ListDataset():  # for training
         if platform == 'darwin':  # macos
             self.img_files = [path.replace('\n', '').replace('/images', '/Users/glennjocher/Downloads/DATA/coco/images')
                               for path in self.img_files]
-        else:
+        else:  # linux (gcp cloud)
             self.img_files = [path.replace('\n', '').replace('/images', '../coco/images') for path in self.img_files]
 
         self.label_files = [path.replace('images', 'labels').replace('.png', '.txt').replace('.jpg', '.txt') for path in