diff --git a/train.py b/train.py
index b724869..27083a5 100644
--- a/train.py
+++ b/train.py
@@ -4,8 +4,9 @@ from datetime import datetime
 from pathlib import Path
 
 from ultralytics import YOLOv10
-from constants import current_images_dir, current_labels_dir, annotation_classes, today_dataset, prefix
+from constants import current_images_dir, current_labels_dir, annotation_classes, prefix
 
+today_dataset = os.path.join('datasets', f'{prefix}{datetime.now():%Y-%m-%d}')
 yaml_path = os.path.join(today_dataset, 'data.yaml')
 train_set = 70
 valid_set = 20