Updated code example to match comment.