Fix the tests I broke to check the output.