Simplify unit test