Small fix after code review.