diff --git a/api/docker/entrypoint.sh b/api/docker/entrypoint.sh
index 9eda9c9b0..e823434f3 100755
--- a/api/docker/entrypoint.sh
+++ b/api/docker/entrypoint.sh
@@ -5,6 +5,8 @@ set -e
 if [[ "${MIGRATION_ENABLED}" == "true" ]]; then
   echo "Running migrations"
   flask upgrade-db
+  echo "Running migrations(dify_plus extend)"
+  flask extend_db upgrade
 fi
 
 if [[ "${MODE}" == "worker" ]]; then