update the test to show the upgrade feature