diff --git a/treeview/src/main/kotlin/io/github/dingyi222666/view/treeview/TreeView.kt b/treeview/src/main/kotlin/io/github/dingyi222666/view/treeview/TreeView.kt
index 474a559..6e70a6e 100644
--- a/treeview/src/main/kotlin/io/github/dingyi222666/view/treeview/TreeView.kt
+++ b/treeview/src/main/kotlin/io/github/dingyi222666/view/treeview/TreeView.kt
@@ -501,7 +501,7 @@ class TreeView<T : Any>(context: Context, attrs: AttributeSet?, defStyleAttr: In
             return
         }
         coroutineScope.launch {
-            if (old !== SelectionMode.NONE) {
+            if (new == SelectionMode.SINGLE || new == SelectionMode.NONE) {
                 tree.selectAllNode(false)
             }
             refresh(true)