fix the unit test output