fix comment following code review.