add support for passing in upgrade value