Fix unit test comment