diff --git a/frontend/train_page.py b/frontend/train_page.py
index e5db3c1..27d307f 100644
--- a/frontend/train_page.py
+++ b/frontend/train_page.py
@@ -47,7 +47,7 @@ def train_page():
             dataset = [ds["message"][0] for ds in dataset["dataset_items"]] 
 
             train_model(get_model(), get_tokenizer(),
-                         dataset, get_workdir() + "/1", 
+                         dataset, get_workdir() + "/training" + "/1", 
                          learning_rate, per_device_train_batch_size, epoch,
                            save_steps, lora_rank)