Fixed minor bug with new updater