simplify unit test