diff --git a/frontend/components/document/FormDelete.vue b/frontend/components/document/FormDelete.vue
index e99cdd06..2b7c08df 100644
--- a/frontend/components/document/FormDelete.vue
+++ b/frontend/components/document/FormDelete.vue
@@ -3,7 +3,7 @@
     :items="selected"
     :title="$t('dataset.deleteDocumentsTitle')"
     :message="$t('dataset.deleteDocumentsMessage')"
-    item-key="text"
+    :item-key="itemKey"
     @ok="$emit('remove')"
     @cancel="$emit('cancel')"
   />
@@ -22,6 +22,10 @@ export default Vue.extend({
     selected: {
       type: Array,
       default: () => []
+    },
+    itemKey: {
+      type: String,
+      default: 'text'
     }
   }
 })
diff --git a/frontend/domain/models/project/project.ts b/frontend/domain/models/project/project.ts
index ff698e80..5fb3c6ae 100644
--- a/frontend/domain/models/project/project.ts
+++ b/frontend/domain/models/project/project.ts
@@ -4,7 +4,7 @@ export interface CurrentUsersRole {
   is_annotation_approver: boolean;
 }
 
-export type ProjectType = 'DocumentClassification' | 'SequenceLabeling' | 'Seq2seq'
+export type ProjectType = 'DocumentClassification' | 'SequenceLabeling' | 'Seq2seq' | 'ImageClassification'
 
 
 export class ProjectReadItem {
diff --git a/frontend/pages/projects/_id/dataset/index.vue b/frontend/pages/projects/_id/dataset/index.vue
index a1260864..b5ee5b50 100644
--- a/frontend/pages/projects/_id/dataset/index.vue
+++ b/frontend/pages/projects/_id/dataset/index.vue
@@ -25,6 +25,7 @@
       <v-dialog v-model="dialogDelete">
         <form-delete
           :selected="selected"
+          :item-key="itemKey"
           @cancel="dialogDelete=false"
           @remove="remove"
         />
@@ -42,7 +43,7 @@
       </v-dialog>
     </v-card-title>
     <image-list
-      v-if="project.projectType === 'ImageClassification'"
+      v-if="isImageTask"
       v-model="selected"
       :items="item.items"
       :is-loading="isLoading"
@@ -110,6 +111,16 @@ export default Vue.extend({
     },
     projectId(): string {
       return this.$route.params.id
+    },
+    isImageTask(): boolean {
+      return this.project.projectType === 'ImageClassification'
+    },
+    itemKey(): string {
+      if (this.isImageTask) {
+        return 'filename'
+      } else {
+        return 'text'
+      }
     }
   },