diff --git a/api/models/model.py b/api/models/model.py
index 3d208d948b40b9..e4652d50ca00da 100644
--- a/api/models/model.py
+++ b/api/models/model.py
@@ -18,15 +18,12 @@
 from core.file.tool_file_parser import ToolFileParser
 from libs.helper import generate_string
 from models.enums import CreatedByRole
-from models.workflow import WorkflowRunStatus
+from models.workflow import WorkflowRunStatus, Workflow
 
 from .account import Account, Tenant
 from .engine import db
 from .types import StringUUID
 
-if TYPE_CHECKING:
-    from .workflow import Workflow
-
 
 class DifySetup(db.Model):
     __tablename__ = "dify_setups"