Fixed tests by using the migration directly.