Update test result expectations for TreeWalker