chore(runway): cherry-pick fix: small refactoring of the latest migration script + add a new migration case

chore(runway): cherry-pick fix: small refactoring of the latest migration script + add a new migration case #76226