Fix bug with updater