Update TreeWalker.as