add correct callback to getUpgrade