Minor changes to test output