Clean up output for tests