Fix updater bug