update tests for changed upgrade command