Minor implementation change.

Per code review comment.