Improve code coverage in TreeNode