Draft: small fix in Draft Upgrade