diff --git a/Code/1_data_prepare/1_2_split_dataset.py b/Code/1_data_prepare/1_2_split_dataset.py
index da01167351b439f20253b4788b8ad95c354bf994..1e20fb61af11bc66ee1c798a12f6e8231ecf3095 100644
--- a/Code/1_data_prepare/1_2_split_dataset.py
+++ b/Code/1_data_prepare/1_2_split_dataset.py
@@ -27,7 +27,7 @@ if __name__ == '__main__':
 
     for root, dirs, files in os.walk(dataset_dir):
         for sDir in dirs:
-            imgs_list = glob.glob(os.path.join(root, sDir) + '*.png')
+            imgs_list = glob.glob(os.path.join(root, sDir, '*.png'))
             random.seed(666)
             random.shuffle(imgs_list)
             imgs_num = len(imgs_list)