Merge pull request #310 from Ambrevar/v2FixWalkNoContainer

v2: Fix walk with non-container root nodes
2016-10-08 18:24:09 +03:00 · 2016-10-08 18:24:09 +03:00 · 2b483a8555
parent 6141d5fde1 14a0c487b8
commit 2b483a8555
1 changed files with 1 additions and 1 deletions
--- a/node.go
+++ b/node.go
@ -309,7 +309,7 @@ func newNodeWalker(root *Node) *nodeWalker {
 }

 func (nw *nodeWalker) next() {
-	if !nw.entering && nw.current == nw.root {
+	if (!nw.current.isContainer() || !nw.entering) && nw.current == nw.root {
 		nw.current = nil
 		return
 	}