diff --git a/0_3_General_modelization_dataset.py b/0_3_General_modelization_dataset.py
index 2feb2a0..a79ea72 100644
--- a/0_3_General_modelization_dataset.py
+++ b/0_3_General_modelization_dataset.py
@@ -21,7 +21,7 @@ warnings.filterwarnings('ignore')
 
 # functions
 def generate_test_set(type_of_comp):
-    file_path_list = fs.ls(f"projet-bdc2324-team1/Generalization/{type_of_comp}/Test_set")
+    file_path_list = fs.ls(f"projet-bdc2324-team1/Generalization_v2/{type_of_comp}/Test_set")
     test_set = pd.DataFrame()
     for file in file_path_list:
         print(file)
@@ -32,7 +32,7 @@ def generate_test_set(type_of_comp):
 
 
 def generate_train_set(type_of_comp):
-    file_path_list = fs.ls(f"projet-bdc2324-team1/Generalization/{type_of_comp}/Train_set")
+    file_path_list = fs.ls(f"projet-bdc2324-team1/Generalization_v2/{type_of_comp}/Train_set")
     train_set = pd.DataFrame()
     for file in file_path_list:
         print(file)
@@ -43,7 +43,7 @@ def generate_train_set(type_of_comp):
 
 
 type_of_comp = input('Choisissez le type de compagnie : sport ? musique ? musee ?')
-BUCKET_OUT = f'projet-bdc2324-team1/Generalization/{type_of_comp}/'
+BUCKET_OUT = f'projet-bdc2324-team1/Generalization_v2/{type_of_comp}/'
 
 # create test and train datasets
 test_set = generate_test_set(type_of_comp)