Fix problem in test_migrations