Minor fix according to the review.