Small fixes in updater