updating code example to match codebase