diff --git a/api/migrations/versions/2024_09_29_0835-ddcc8bbef391_increase_max_length_of_builtin_tool_provider.py b/api/migrations/versions/2024_09_29_0835-ddcc8bbef391_increase_max_length_of_builtin_tool_provider.py
index 5d3ab7f9ad..d7a5e4b1ea 100644
--- a/api/migrations/versions/2024_09_29_0835-ddcc8bbef391_increase_max_length_of_builtin_tool_provider.py
+++ b/api/migrations/versions/2024_09_29_0835-ddcc8bbef391_increase_max_length_of_builtin_tool_provider.py
@@ -12,7 +12,7 @@ import sqlalchemy as sa
 
 # revision identifiers, used by Alembic.
 revision = 'ddcc8bbef391'
-down_revision = 'cf8f4fc45278' # HEAD OF PLUGIN BRANCH
+down_revision = 'd7999dfa4aae' # HEAD OF PLUGIN BRANCH
 branch_labels = None
 depends_on = None
 
diff --git a/api/models/__init__.py b/api/models/__init__.py
index b0b9880ca4..1237c55a30 100644
--- a/api/models/__init__.py
+++ b/api/models/__init__.py
@@ -73,7 +73,6 @@ from .task import CeleryTask, CeleryTaskSet
 from .tools import (
     ApiToolProvider,
     BuiltinToolProvider,
-    PublishedAppTool,
     ToolConversationVariables,
     ToolFile,
     ToolLabelBinding,
@@ -150,7 +149,6 @@ __all__ = [
     "ProviderOrder",
     "ProviderQuotaType",
     "ProviderType",
-    "PublishedAppTool",
     "RecommendedApp",
     "SavedMessage",
     "Site",
diff --git a/api/models/base.py b/api/models/base.py
index da4648efa6..da9509301a 100644
--- a/api/models/base.py
+++ b/api/models/base.py
@@ -1,5 +1,5 @@
 from sqlalchemy.orm import declarative_base
 
-from extensions.ext_database import metadata
+from models.engine import metadata
 
 Base = declarative_base(metadata=metadata)