Minor update from code review.