diff --git a/lib/tree.rb b/lib/tree.rb index c841b4e..9290425 100644 --- a/lib/tree.rb +++ b/lib/tree.rb @@ -849,7 +849,7 @@ def print_tree(level = 0, max_depth = nil, block = lambda { |node, prefix| puts return unless max_depth.nil? || level < max_depth # Exit if the max level is defined, and reached. - children { |child| child.print_tree(level + 1, block) if child } # Child might be 'nil' + children { |child| child.print_tree(level + 1, max_depth, block) if child } # Child might be 'nil' end end