Updating the traversal impl