fix migration_test