Simplify the testing code.