Bugfix for tree depth argument