Fix up some alembic stuff