diff --git a/lib/yarp.rb b/lib/yarp.rb index 48b1f173dd..28f0fae003 100644 --- a/lib/yarp.rb +++ b/lib/yarp.rb @@ -520,7 +520,7 @@ module YARP locals << [] if node.once? end - stack.concat(node.child_nodes.compact) + stack.concat(node.compact_child_nodes) end locals diff --git a/lib/yarp/pattern.rb b/lib/yarp/pattern.rb index 339f1506c3..f7519137e4 100644 --- a/lib/yarp/pattern.rb +++ b/lib/yarp/pattern.rb @@ -73,7 +73,7 @@ module YARP while (node = queue.shift) yield node if @compiled.call(node) - queue.concat(node.child_nodes.compact) + queue.concat(node.compact_child_nodes) end end diff --git a/test/yarp/parse_serialize_test.rb b/test/yarp/parse_serialize_test.rb index d3474f7104..18ff0c4319 100644 --- a/test/yarp/parse_serialize_test.rb +++ b/test/yarp/parse_serialize_test.rb @@ -31,7 +31,7 @@ module YARP while (node = queue.shift) return node if node.is_a?(SourceFileNode) - queue.concat(node.child_nodes.compact) + queue.concat(node.compact_child_nodes) end end end diff --git a/test/yarp/parse_test.rb b/test/yarp/parse_test.rb index 0b470f6eed..eba5767776 100644 --- a/test/yarp/parse_test.rb +++ b/test/yarp/parse_test.rb @@ -204,7 +204,7 @@ module YARP queue = [program] while (node = queue.shift) return node if node.is_a?(SourceFileNode) - queue.concat(node.child_nodes.compact) + queue.concat(node.compact_child_nodes) end end