Updater code fix