diff --git a/components/tree-select/index.jsx b/components/tree-select/index.jsx
index 1a00fb6a0..29f9d4e3e 100644
--- a/components/tree-select/index.jsx
+++ b/components/tree-select/index.jsx
@@ -58,7 +58,7 @@ export default {
           // label: newLabel,
           title: newTitle || newLabel,
         }
-        this.updateTreeData(children)
+        this.updateTreeData(children || [])
         Object.assign(list[i], item)
       }
     },
@@ -72,7 +72,7 @@ export default {
         dropdownClassName,
         ...restProps
       } = props
-      this.updateTreeData(props.treeData)
+      this.updateTreeData(props.treeData || [])
       const cls = {
         [`${prefixCls}-lg`]: size === 'large',
         [`${prefixCls}-sm`]: size === 'small',