fix test broken by ux changes