Updated unit test comment