Fix a bug in test_alembic