diff --git a/train_sumo_ppo.py b/train_sumo_ppo.py
index 8a71ecc..6fe57f0 100644
--- a/train_sumo_ppo.py
+++ b/train_sumo_ppo.py
@@ -29,8 +29,8 @@ def train_sumo_ppo():
 
     start_episode = 1
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-    checkpoint_dir = os.path.join(train_config["checkpoint_dir"], timestamp)
-    log_dir = os.path.join(train_config["log_dir"], timestamp)
+    checkpoint_dir = os.path.join("checkpoints_sumo_ppo", timestamp)
+    log_dir = os.path.join("logs_sumo_ppo", timestamp)
     os.makedirs(checkpoint_dir, exist_ok=True)
     os.makedirs(log_dir, exist_ok=True)
     with open(os.path.join(checkpoint_dir, "config.yaml"), "w", encoding="utf-8") as f: