Simplify unit tests.