Minor fixes to updater