chore(test): adding upgrade in task