Add support for the migration flags.