diff --git a/api/controllers/console/auth/data_source_bearer_auth.py b/api/controllers/console/auth/data_source_bearer_auth.py
index dd6f631a11e9b0..38ab6828b92d89 100644
--- a/api/controllers/console/auth/data_source_bearer_auth.py
+++ b/api/controllers/console/auth/data_source_bearer_auth.py
@@ -1,6 +1,7 @@
 from flask_login import current_user
 from flask_restful import Resource, reqparse
 from werkzeug.exceptions import Forbidden
+from controllers.console.auth.error import ApiKeyAuthFailedError
 
 from controllers.console import api
 from libs.login import login_required