updating code example to match codebase again