Fix bug in updater