Refactor test to actually check output.