Fix alembic upgrade script