refactor test
refactor ut