diff --git a/api/controllers/inner_api/workspace/workspace.py b/api/controllers/inner_api/workspace/workspace.py index 436854698e..9dfa5d23c3 100644 --- a/api/controllers/inner_api/workspace/workspace.py +++ b/api/controllers/inner_api/workspace/workspace.py @@ -6,6 +6,7 @@ from controllers.console.wraps import setup_required from controllers.inner_api import api from controllers.inner_api.wraps import enterprise_inner_api_only from events.tenant_event import tenant_was_created +from extensions.ext_database import db from models.account import Account from services.account_service import TenantService @@ -19,7 +20,7 @@ class EnterpriseWorkspace(Resource): parser.add_argument("owner_email", type=str, required=True, location="json") args = parser.parse_args() - account = Account.query.filter_by(email=args["owner_email"]).first() + account = db.session.query(Account).filter_by(email=args["owner_email"]).first() if account is None: return {"message": "owner account not found."}, 404