diff --git a/confic.py b/confic.py
index bc446aa..e765147 100644
--- a/confic.py
+++ b/confic.py
@@ -19,6 +19,6 @@ DATA_DIR = 'data/dataset'
 OUTDIR = 'model_outputs'
 INFERENCE_OUTDIR = 'inference_outputs'
 
-for required_folders in [TRAIN_DIR, OUTDIR, INFERENCE_OUTDIR]:
+for required_folders in [DATA_DIR, OUTDIR, INFERENCE_OUTDIR]:
     if not pathlib.Path(required_folders).exists():
         pathlib.Path(required_folders).mkdir(parents=True, exist_ok=True)
diff --git a/datasets.py b/datasets.py
index bcc6270..8fc3020 100644
--- a/datasets.py
+++ b/datasets.py
@@ -64,8 +64,7 @@ def create_train_or_test_dataset(path, train=True):
     else:
         print('Generate test dataset !')
         pfx='test'
-    embed()
-    quit()
+
     csv_candidates = list(Path(path).rglob(f'*{pfx}*.csv'))
     if len(csv_candidates) == 0:
         print(f'no .csv files for *{pfx}* found in {Path(path)}')