Small change following code review.