tests: update UT

update tests