Fix up the migration test