refactor test code
refactor test code