diff --git a/0_2_Dataset_construction.py b/0_2_Dataset_construction.py
index 2e9b9e0..26e3ad3 100644
--- a/0_2_Dataset_construction.py
+++ b/0_2_Dataset_construction.py
@@ -147,7 +147,7 @@ for company in list_of_comp:
 
     # Exportation
     FILE_KEY_OUT_S3 = "dataset_test" + company +  ".csv"
-    FILE_PATH_OUT_S3 = BUCKET_OUT + "/" + FILE_KEY_OUT_S3
+    FILE_PATH_OUT_S3 = BUCKET_OUT + "/Test_set/" + FILE_KEY_OUT_S3
     
     with fs.open(FILE_PATH_OUT_S3, 'w') as file_out:
         dataset_test.to_csv(file_out, index = False)
@@ -159,7 +159,7 @@ for company in list_of_comp:
                                         max_date = final_date, directory_path = company)
     # Export
     FILE_KEY_OUT_S3 = "dataset_train" + company + ".csv" 
-    FILE_PATH_OUT_S3 = BUCKET_OUT + "/" + FILE_KEY_OUT_S3
+    FILE_PATH_OUT_S3 = BUCKET_OUT + "/Train_test/" + FILE_KEY_OUT_S3
     
     with fs.open(FILE_PATH_OUT_S3, 'w') as file_out:
         dataset_train.to_csv(file_out, index = False)