Fixed treeNode return statements.