Update for migration tests.