Fix to alembic script