Fix bug I introduced in test_output.py