diff --git a/components/tree/TreeIndent.razor b/components/tree/TreeIndent.razor
index 8e83903c..a4336091 100644
--- a/components/tree/TreeIndent.razor
+++ b/components/tree/TreeIndent.razor
@@ -1,4 +1,5 @@
 @namespace AntDesign
+@inherits ComponentBase
 @typeparam TItem
 
 <span class="ant-tree-indent" aria-hidden="true">
diff --git a/components/tree/TreeIndent.razor.cs b/components/tree/TreeIndent.razor.cs
index 01b42f75..a9b27adb 100644
--- a/components/tree/TreeIndent.razor.cs
+++ b/components/tree/TreeIndent.razor.cs
@@ -5,7 +5,7 @@ using Microsoft.AspNetCore.Components;
 
 namespace AntDesign
 {
-    public partial class TreeIndent<TItem>
+    public partial class TreeIndent<TItem> : ComponentBase
     {
         /// <summary>
         /// 树控件本身
@@ -21,7 +21,5 @@ namespace AntDesign
 
         [Parameter]
         public int TreeLevel { get; set; }
-
-
     }
 }
diff --git a/components/tree/TreeNode.razor b/components/tree/TreeNode.razor
index 63a459c2..4fac3859 100644
--- a/components/tree/TreeNode.razor
+++ b/components/tree/TreeNode.razor
@@ -5,7 +5,7 @@
 @if (RealDisplay)
 {
     <CascadingValue Value="this" Name="SelfNode">
-        <div @key="NodeId" class="@TreeNodeClassMapper.Class">
+        <div @key="NodeId" class="@ClassMapper.Class" style="@Style">
 
             <TreeIndent TreeLevel="@TreeLevel" TItem="TItem"></TreeIndent>
 
@@ -20,7 +20,6 @@
             }
 
             <TreeNodeTitle TItem="TItem"></TreeNodeTitle>
-
         </div>
     </CascadingValue>
 }
@@ -36,4 +35,4 @@
     {
         @Nodes
     }
-</CascadingValue>
+</CascadingValue>
\ No newline at end of file
diff --git a/components/tree/TreeNode.razor.cs b/components/tree/TreeNode.razor.cs
index 6e18917e..7df1ad8f 100644
--- a/components/tree/TreeNode.razor.cs
+++ b/components/tree/TreeNode.razor.cs
@@ -188,11 +188,9 @@ namespace AntDesign
         [Parameter]
         public bool Loading { get; set; }
 
-        protected ClassMapper TreeNodeClassMapper { get; } = new ClassMapper();
-
         private void SetTreeNodeClassMapper()
         {
-            TreeNodeClassMapper.Clear().Add("ant-tree-treenode")
+            ClassMapper.Clear().Add("ant-tree-treenode")
                 .If("ant-tree-treenode-disabled", () => Disabled)
                 .If("ant-tree-treenode-switcher-open", () => SwitcherOpen)
                 .If("ant-tree-treenode-switcher-close", () => SwitcherClose)
diff --git a/components/tree/TreeNodeCheckbox.razor b/components/tree/TreeNodeCheckbox.razor
index 15cc98de..6fa2fa53 100644
--- a/components/tree/TreeNodeCheckbox.razor
+++ b/components/tree/TreeNodeCheckbox.razor
@@ -1,10 +1,8 @@
 @namespace AntDesign
-@inherits AntDomComponentBase
+@inherits ComponentBase
 @typeparam TItem
 
-    <span class="@ClassMapper.Class" @onclick="OnClick">
-        <span class="ant-tree-checkbox-inner">
-
-        </span>
-
+<span class="@ClassMapper.Class" @onclick="OnClick">
+    <span class="ant-tree-checkbox-inner">
     </span>
+</span>
\ No newline at end of file
diff --git a/components/tree/TreeNodeCheckbox.razor.cs b/components/tree/TreeNodeCheckbox.razor.cs
index cae8f1cd..8e418ded 100644
--- a/components/tree/TreeNodeCheckbox.razor.cs
+++ b/components/tree/TreeNodeCheckbox.razor.cs
@@ -4,7 +4,7 @@ using Microsoft.AspNetCore.Components.Web;
 
 namespace AntDesign
 {
-    public partial class TreeNodeCheckbox<TItem> : AntDomComponentBase
+    public partial class TreeNodeCheckbox<TItem> : ComponentBase
     {
         /// <summary>
         /// 树控件本身
diff --git a/components/tree/TreeNodeSwitcher.razor b/components/tree/TreeNodeSwitcher.razor
index 3adf8fc1..ca0ce12c 100644
--- a/components/tree/TreeNodeSwitcher.razor
+++ b/components/tree/TreeNodeSwitcher.razor
@@ -1,5 +1,5 @@
 @namespace AntDesign
-@inherits AntDomComponentBase
+@inherits ComponentBase
 @typeparam TItem
 
 <span class="@ClassMapper.Class" @onclick="OnClick">
diff --git a/components/tree/TreeNodeSwitcher.razor.cs b/components/tree/TreeNodeSwitcher.razor.cs
index afa2023d..855b6d8a 100644
--- a/components/tree/TreeNodeSwitcher.razor.cs
+++ b/components/tree/TreeNodeSwitcher.razor.cs
@@ -8,7 +8,7 @@ using Microsoft.AspNetCore.Components.Web;
 
 namespace AntDesign
 {
-    public partial class TreeNodeSwitcher<TItem> : AntDomComponentBase
+    public partial class TreeNodeSwitcher<TItem> : ComponentBase
     {
         /// <summary>
         /// 树控件本身
diff --git a/components/tree/TreeNodeTitle.razor b/components/tree/TreeNodeTitle.razor
index 5cc9d63e..1a071aa3 100644
--- a/components/tree/TreeNodeTitle.razor
+++ b/components/tree/TreeNodeTitle.razor
@@ -1,5 +1,5 @@
 @namespace AntDesign
-@inherits AntDomComponentBase
+@inherits ComponentBase
 @typeparam TItem
 
 <span class="@TitleClassMapper.Class" title="@SelfNode.Title" @onclick="OnClick" @ondblclick="OnDblClick" draggable="@CanDraggable" aria-grabbed="@CanDraggable">
@@ -33,5 +33,4 @@
             @SelfNode.Title
         </span>
     }
-</span>
-
+</span>
\ No newline at end of file
diff --git a/components/tree/TreeNodeTitle.razor.cs b/components/tree/TreeNodeTitle.razor.cs
index 0932ff18..0f672c28 100644
--- a/components/tree/TreeNodeTitle.razor.cs
+++ b/components/tree/TreeNodeTitle.razor.cs
@@ -7,7 +7,7 @@ using Microsoft.AspNetCore.Components.Web;
 
 namespace AntDesign
 {
-    public partial class TreeNodeTitle<TItem> : AntDomComponentBase
+    public partial class TreeNodeTitle<TItem> : ComponentBase
     {
         /// <summary>
         /// 树控件本身