diff --git a/src/components/TreeTable/index.vue b/src/components/TreeTable/index.vue
index 30b3b10e..92ddc434 100644
--- a/src/components/TreeTable/index.vue
+++ b/src/components/TreeTable/index.vue
@@ -149,6 +149,19 @@ export default {
           this.selcetRecursion(child, select, children)
         })
       }
+    },
+    updateTreeNode(item) {
+      return new Promise(resolve => {
+        const { _id, _parent } = item
+        const index = _id.split('-').slice(-1)[0] // get last index
+        if (_parent) {
+          _parent.children.splice(index, 1, item)
+          resolve(this.data)
+        } else {
+          this.data.splice(index, 1, item)
+          resolve(this.data)
+        }
+      })
     }
   }
 }
diff --git a/src/views/tree-table/custom/index.vue b/src/views/tree-table/custom/index.vue
index 89f06058..754ec202 100644
--- a/src/views/tree-table/custom/index.vue
+++ b/src/views/tree-table/custom/index.vue
@@ -129,17 +129,8 @@ export default {
       this.tempItem = Object.assign({}, row)
       this.dialogFormVisible = true
     },
-    updateItem() {
-      const data = this.$refs.TreeTable.getData()
-      const { _id } = this.tempItem
-
-      for (let i = 0; i < data.length; i++) {
-        if (data[i]._id === _id) {
-          data.splice(i, 1, Object.assign({}, this.tempItem))
-          break
-        }
-      }
-
+    async updateItem() {
+      await this.$refs.TreeTable.updateTreeNode(this.tempItem)
       this.dialogFormVisible = false
     },
     addMenuItem(row, type) {