diff --git a/.github/workflows/db-migration-test.yml b/.github/workflows/db-migration-test.yml
index 158be4f5133d80..64e8eb291c466e 100644
--- a/.github/workflows/db-migration-test.yml
+++ b/.github/workflows/db-migration-test.yml
@@ -55,4 +55,4 @@ jobs:
       - name: Run DB Migration
         run: |
           cd api
-          poetry run python -m flask db-migrate
+          poetry run python -m flask upgrade-db
diff --git a/api/docker/entrypoint.sh b/api/docker/entrypoint.sh
index 7e1eb32d3dd625..7b1f1dfc03c679 100755
--- a/api/docker/entrypoint.sh
+++ b/api/docker/entrypoint.sh
@@ -4,7 +4,7 @@ set -e
 
 if [[ "${MIGRATION_ENABLED}" == "true" ]]; then
   echo "Running migrations"
-  flask db-migrate
+  flask upgrade-db
 fi
 
 if [[ "${MODE}" == "worker" ]]; then