diff --git a/datasets/hed/get_hed.py b/datasets/hed/get_hed.py
index 8b61b62..b317d94 100644
--- a/datasets/hed/get_hed.py
+++ b/datasets/hed/get_hed.py
@@ -41,7 +41,7 @@ def get_pretrain_data(labels, image_size=(28, 28, 1)):
     X = []
     for label in labels:
         label_path = os.path.join(
-            "./datasets/hed/dataset/mnist_images", label
+            "./datasets/hed/mnist_images", label
         )
         img_path_list = os.listdir(label_path)
         for img_path in img_path_list: