diff --git a/ui/src/views/log/component/EditContentDialog.vue b/ui/src/views/log/component/EditContentDialog.vue
index d29679735..7b28b9d5b 100644
--- a/ui/src/views/log/component/EditContentDialog.vue
+++ b/ui/src/views/log/component/EditContentDialog.vue
@@ -36,7 +36,7 @@
           <template #default="{ node, data }">
             <span class="flex align-center">
               <AppAvatar
-                v-if="!node.isLeaf && data.type === '1'"
+                v-if="!data.dataset_id && data.type === '1'"
                 class="mr-12 avatar-purple"
                 shape="square"
                 :size="24"
@@ -44,7 +44,7 @@
                 <img src="@/assets/icon_web.svg" style="width: 58%" alt="" />
               </AppAvatar>
               <AppAvatar
-                v-else-if="!node.isLeaf && data.type === '0'"
+                v-else-if="!data.dataset_id && data.type === '0'"
                 class="mr-12"
                 shape="square"
                 :size="24"