diff --git a/train.py b/train.py
index 566ec3c..ae6ef10 100644
--- a/train.py
+++ b/train.py
@@ -100,13 +100,15 @@ if __name__ == '__main__':
     # form_dataset()
     # create_yaml()
 
+    m = latest_model or 'yolov10x.yaml'
+    print(f'Initial model: {m}')
     model = YOLOv10(latest_model or 'yolov10x.yaml')
 
     folder = f'{prefix}2024-06-18'
-    yaml = abspath(path.join('datasets', today_folder, 'data.yaml'))
+    yaml = abspath(path.join('datasets', folder, 'data.yaml'))
     results = model.train(data=yaml, epochs=100, batch=10, imgsz=640)
 
     shutil.copy(f'{results.save_dir}/weights/best.pt', latest_model)
-    shutil.copytree(results.save_dir, f'models/{today_folder}')
+    shutil.copytree(results.save_dir, f'models/{folder}')
     shutil.rmtree('runs')
     shutil.rmtree('models/zombobase-latest')