Updated tree tests to reflect new changes