Tree Node bug fix