diff --git a/ci_cd/run_migration.py b/ci_cd/run_migration.py
index 1cbe9fb59d..feec4046ee 100644
--- a/ci_cd/run_migration.py
+++ b/ci_cd/run_migration.py
@@ -15,7 +15,7 @@ DEFAULT_BASE_BRANCH = "litellm_internal_staging"
 
 
 def _find_destructive_statements(sql: str) -> list:
-    """Return SQL lines containing DROP COLUMN or DROP TABLE."""
+    """Return SQL lines containing DROP COLUMN, DROP TABLE, or DROP INDEX."""
     return [
         line.strip() for line in sql.splitlines() if DESTRUCTIVE_PATTERN.search(line)
     ]
@@ -210,7 +210,8 @@ def create_migration(
         allow_destructive (bool): Required to write a migration that contains
             DROP COLUMN, DROP TABLE, or DROP INDEX statements. Without this
             flag, the script exits non-zero and prints guidance.
-        base_branch (str): Branch to check freshness against (default: "main").
+        base_branch (str): Branch to check freshness against
+            (default: "litellm_internal_staging").
         skip_freshness_check (bool): Skip the "branch is up to date" check.
             Only for intentional migrations against an older base.
     """