fix upgrading dialog