Fix alembic test