Improve the test output