From e7c59d145ca4d761622803f5db18abeb368334ef Mon Sep 17 00:00:00 2001 From: Evan Sharp Date: Thu, 23 Oct 2014 16:17:03 -0400 Subject: [PATCH] Correctly recurse print_tree method --- lib/tree.rb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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