diff --git a/packages/tree/src/model/tree-store.js b/packages/tree/src/model/tree-store.js
index 5c96e5ceb..d0bcd3ea5 100644
--- a/packages/tree/src/model/tree-store.js
+++ b/packages/tree/src/model/tree-store.js
@@ -327,8 +327,8 @@ export default class TreeStore {
   }
 
   setCurrentNodeKey(key) {
-    if (key === null) {
-      this.currentNode.isCurrent = false;
+    if (key === null || key === undefined) {
+      this.currentNode && (this.currentNode.isCurrent = false);
       this.currentNode = null;
       return;
     }