Fix display error for TreeView