Improve test logging