diff --git a/desktop/plugins/public/ui-debugger/components/Tree2.tsx b/desktop/plugins/public/ui-debugger/components/Tree2.tsx
index 80e72a392..50ca804b3 100644
--- a/desktop/plugins/public/ui-debugger/components/Tree2.tsx
+++ b/desktop/plugins/public/ui-debugger/components/Tree2.tsx
@@ -88,7 +88,7 @@ export function Tree2({
           instance.uiState.hoveredNodes.set([]);
         }}>
         {treeNodes.map((treeNode, index) => (
-          <TreeItemContainer
+          <MemoTreeItemContainer
             innerRef={refs[index]}
             isUsingKBToScroll={isUsingKBToScroll}
             key={treeNode.id}
@@ -107,6 +107,18 @@ export type TreeNode = UINode & {
   isExpanded: boolean;
 };
 
+const MemoTreeItemContainer = React.memo(
+  TreeItemContainer,
+  (prevProps, nextProps) => {
+    const id = prevProps.treeNode.id;
+    return (
+      prevProps.treeNode === nextProps.treeNode &&
+      id !== prevProps.selectedNode &&
+      id !== nextProps.selectedNode
+    );
+  },
+);
+
 function TreeItemContainer({
   innerRef,
   isUsingKBToScroll,