diff --git a/packages/theme/src/tree/index.less b/packages/theme/src/tree/index.less
index 099c720374..37e209ab28 100644
--- a/packages/theme/src/tree/index.less
+++ b/packages/theme/src/tree/index.less
@@ -367,9 +367,11 @@
         fill: var(--tv-Tree-node-checked-icon-color);
       }
     }
-    input {
-      height: calc(var(--tv-Tree-node-content-height) - 2px);
-      font-size: var(--tv-Tree-node-label-font-size);
+    .@{tree-node-prefix-cls}__edit-input {
+      input {
+        height: calc(var(--tv-Tree-node-content-height) - 2px);
+        font-size: var(--tv-Tree-node-label-font-size);
+      }
     }
   }
 
diff --git a/packages/vue/src/tree/src/tree-node.vue b/packages/vue/src/tree/src/tree-node.vue
index e0c29d896e..b68aa1ad15 100644
--- a/packages/vue/src/tree/src/tree-node.vue
+++ b/packages/vue/src/tree/src/tree-node.vue
@@ -121,6 +121,7 @@
           <template v-if="action.type === 'edit' && action.node && action.node === node">
             <tiny-input
               v-model="action.label"
+              class="tiny-tree-node__edit-input"
               ref="editInput"
               :autofocus="true"
               clearable