diff --git a/src/services/task_service.py b/src/services/task_service.py
index 0537e933..6fca5461 100644
--- a/src/services/task_service.py
+++ b/src/services/task_service.py
@@ -1,6 +1,7 @@
 import asyncio
 import random
 from typing import Dict, Optional
+import uuid
 
 from models.tasks import TaskStatus, UploadTask, FileTask
 from utils.gpu_detection import get_worker_count