Fix examples of TreeWalker module.