Fix upgrade for MSI