oops
refactor test cases.