diff --git a/app/api/endpoints/workflow.py b/app/api/endpoints/workflow.py
index cbf58d77..24f948b8 100644
--- a/app/api/endpoints/workflow.py
+++ b/app/api/endpoints/workflow.py
@@ -5,9 +5,9 @@ from sqlalchemy.orm import Session
 
 from app import schemas
 from app.db import get_db
+from app.db.models.workflow import Workflow
 from app.db.user_oper import get_current_active_superuser, get_current_active_user
 from chain.workflow import WorkflowChain
-from db.models.workflow import Workflow
 
 router = APIRouter()
 
diff --git a/app/db/models/__init__.py b/app/db/models/__init__.py
index 235d3b06..09ecf69f 100644
--- a/app/db/models/__init__.py
+++ b/app/db/models/__init__.py
@@ -8,3 +8,4 @@ from .systemconfig import SystemConfig
 from .transferhistory import TransferHistory
 from .user import User
 from .userconfig import UserConfig
+from .workflow import Workflow