diff --git a/services/ml/app/main.py b/services/ml/app/main.py
index f54cf82..588fbab 100644
--- a/services/ml/app/main.py
+++ b/services/ml/app/main.py
@@ -10,7 +10,7 @@ import logging
 import os
 import re
 import threading
-import uuid as uuid_lib
+import uuid
 from pathlib import Path
 from typing import Optional, Dict, Any, List
 from datetime import datetime
@@ -1341,7 +1341,7 @@ async def training_start(request: TrainingStartRequest):
                     "run_id": state.active_training_run_id,
                 },
             )
-        run_id = str(uuid_lib.uuid4())
+        run_id = str(uuid.uuid4())
         state.active_training_run_id = run_id
 
     config = state.pipeline_config or get_default_config()