diff --git a/image.py b/image.py
index c8506d49..72a5cc44 100644
--- a/image.py
+++ b/image.py
@@ -7,7 +7,7 @@
 import cv2
 
 def load_data(img_path,train = True):
-    gt_path = img_path.replace('.jpg','.h5').replace('images','ground_truth')
+    gt_path = img_path.replace('.jpg','.h5').replace('images','ground-truth')
     img = Image.open(img_path).convert('RGB')
     gt_file = h5py.File(gt_path)
     target = np.asarray(gt_file['density'])
@@ -40,4 +40,4 @@ def load_data(img_path,train = True):
     target = cv2.resize(target,(target.shape[1]/8,target.shape[0]/8),interpolation = cv2.INTER_CUBIC)*64
     
     
-    return img,target
\ No newline at end of file
+    return img,target