Fix argument for upgrade call