diff --git a/core/components/AiAddressBookMenu.vue b/core/components/AiAddressBookMenu.vue
index eaf71f9e..b9cd6c4a 100644
--- a/core/components/AiAddressBookMenu.vue
+++ b/core/components/AiAddressBookMenu.vue
@@ -20,9 +20,11 @@
           <el-tree
             ref="tree"
             node-key="id"
+            highlight-current
             :filter-node-method="handleTreeFilter"
             :props="defaultProps" :data="list"
             :default-expanded-keys="treeRoot"
+            :current-node-key="treeRoot"
             @current-change="v=>selected=v">
             <div class="tree-container" slot-scope="{ data }">
               <span>{{ data.name }}</span>