Minor changes to UTS benchmark