Simplify unittest.