simplify unittest