diff --git a/api/core/app/apps/workflow/generate_task_pipeline_fast.py b/api/core/app/apps/workflow/generate_task_pipeline_fast.py
index 0bd6fe3ed8..20ecd29226 100644
--- a/api/core/app/apps/workflow/generate_task_pipeline_fast.py
+++ b/api/core/app/apps/workflow/generate_task_pipeline_fast.py
@@ -1,5 +1,6 @@
 import logging
 import time
+from datetime import UTC, datetime
 from collections.abc import Generator
 from typing import Optional, Union