diff --git a/helper_scripts/Dataset_split.py b/helper_scripts/Dataset_split.py
index 8bea9af..6a2902e 100644
--- a/helper_scripts/Dataset_split.py
+++ b/helper_scripts/Dataset_split.py
@@ -6,8 +6,8 @@
 import shutil
 import random
 
-source_folder = '../data/2DIR'
-testA_folder = '../data/valA'
+source_folder = './data/2DIR'
+testA_folder = './data/valA'
 
 file_list = os.listdir(source_folder)