Fix upgrade