Rework migrate.test.js