Make upgrade dialog a bit more user-friendly