attemping to conver the upgrade case with integration tests