Small fix following code review.