Add alembic fix