diff --git a/make_synthetic_test_dataset.py b/make_synthetic_test_dataset.py
index 64e9c9d..949d080 100644
--- a/make_synthetic_test_dataset.py
+++ b/make_synthetic_test_dataset.py
@@ -26,7 +26,7 @@
 
 DATASET_SIZE = 20000
 RESUME = False
-dataset_dir = Path("./datasets/synthetic-mist-tess-dataset-test/")
+dataset_dir = Path("./datasets/synthetic-mist-tess-dataset/")
 dataset_dir.mkdir(parents=True, exist_ok=True)
 
 # TODO: better way to share inst over multiple calls to generate_instances_from_mist_models()
diff --git a/make_training_dataset.py b/make_training_dataset.py
index 61c258e..95448bc 100644
--- a/make_training_dataset.py
+++ b/make_training_dataset.py
@@ -22,7 +22,7 @@
 from ebop_maven.libs.tee import Tee
 
 DATASET_SIZE = 250000
-dataset_dir = Path(f"./datasets/formal-training-dataset-{DATASET_SIZE // 1000}k-test3/")
+dataset_dir = Path(f"./datasets/formal-training-dataset-{DATASET_SIZE // 1000}k/")
 dataset_dir.mkdir(parents=True, exist_ok=True)