Updater bug fix