Fix unit test logging