Fix upgrade screen