diff --git a/detect.py b/detect.py
index 4cc532844f..2c9899ee49 100644
--- a/detect.py
+++ b/detect.py
@@ -45,7 +45,7 @@
 
 model.eval() # Set in evaluation mode
 
-dataloader = DataLoader(ImageFolder('data/samples', img_size=opt.img_size),
+dataloader = DataLoader(ImageFolder(opt.image_folder, img_size=opt.img_size),
                         batch_size=opt.batch_size, shuffle=False, num_workers=opt.n_cpu)
 
 classes = load_classes(opt.class_path) # Extracts class labels from file