Add support for migration checks.