Fixed comment as pr review.