Updated to match new Updater