diff --git a/frontend/components/containers/documents/DocumentDeletionButton.vue b/frontend/components/containers/documents/DocumentDeletionButton.vue
index dcff1b3a..976d142a 100644
--- a/frontend/components/containers/documents/DocumentDeletionButton.vue
+++ b/frontend/components/containers/documents/DocumentDeletionButton.vue
@@ -1,21 +1,41 @@
 <template>
-  <confirm-dialog
-    title="Delete Document"
-    message="Are you sure you want to delete these documents from this project?"
-    item-key="text"
-    :disabled="!isDocumentSelected"
-    :items="selected"
-    @ok="handleDeleteDocument()"
-  />
+  <div>
+    <v-btn
+      class="text-capitalize"
+      outlined
+      :disabled="!isDocumentSelected"
+      @click="dialog=true"
+    >
+      Delete
+    </v-btn>
+    <base-dialog :dialog="dialog">
+      <confirm-form
+        title="Delete Document"
+        message="Are you sure you want to delete these documents from this project?"
+        item-key="text"
+        :items="selected"
+        @ok="deleteDocument($route.params.id);dialog=false"
+        @cancel="dialog=false"
+      />
+    </base-dialog>
+  </div>
 </template>
 
 <script>
 import { mapState, mapGetters, mapActions } from 'vuex'
-import ConfirmDialog from '@/components/organisms/utils/ConfirmDialog'
+import BaseDialog from '@/components/molecules/BaseDialog'
+import ConfirmForm from '@/components/organisms/utils/ConfirmForm'
 
 export default {
   components: {
-    ConfirmDialog
+    BaseDialog,
+    ConfirmForm
+  },
+
+  data() {
+    return {
+      dialog: false
+    }
   },
 
   computed: {
diff --git a/frontend/pages/projects/_id/dataset/index.vue b/frontend/pages/projects/_id/dataset/index.vue
index 76eaa919..b7e80dbc 100644
--- a/frontend/pages/projects/_id/dataset/index.vue
+++ b/frontend/pages/projects/_id/dataset/index.vue
@@ -1,56 +1,17 @@
 <template>
   <v-card>
     <v-card-title class="mb-2">
-      <action-menu
-        :items="menuItems"
-        @upload="importDialog=true"
-        @download="exportDialog=true"
-      />
-      <base-dialog :dialog="importDialog">
-        <document-upload-form
-          :upload-document="uploadDocument"
-          :formats="formatList"
-          @close="importDialog=false"
-        />
-      </base-dialog>
-      <base-dialog :dialog="exportDialog">
-        <document-export-form
-          :export-document="exportDocument"
-          :formats="['json']"
-          @close="exportDialog=false"
-        />
-      </base-dialog>
-      <v-btn
-        class="text-capitalize ms-2"
-        outlined
-        :disabled="!isDocumentSelected"
-        @click="deleteDialog=true"
-      >
-        Delete
-      </v-btn>
-      <base-dialog :dialog="deleteDialog">
-        <confirm-form
-          title="Delete Document"
-          message="Are you sure you want to delete these documents from this project?"
-          item-key="text"
-          :items="selected"
-          @ok="deleteDocument($route.params.id);deleteDialog=false"
-          @cancel="deleteDialog=false"
-        />
-      </base-dialog>
+      <document-action-menu />
+      <document-deletion-button class="ms-2" />
     </v-card-title>
     <document-list />
   </v-card>
 </template>
 
 <script>
-import { mapActions, mapGetters, mapState } from 'vuex'
-import ActionMenu from '@/components/molecules/ActionMenu'
-import BaseDialog from '@/components/molecules/BaseDialog'
-import ConfirmForm from '@/components/organisms/utils/ConfirmForm'
 import DocumentList from '@/components/containers/documents/DocumentList'
-import DocumentUploadForm from '@/components/organisms/documents/DocumentUploadForm'
-import DocumentExportForm from '@/components/organisms/documents/DocumentExportForm'
+import DocumentActionMenu from '@/components/containers/documents/DocumentActionMenu'
+import DocumentDeletionButton from '@/components/containers/documents/DocumentDeletionButton'
 
 export default {
   layout: 'project',
@@ -58,33 +19,9 @@ export default {
   middleware: 'check-auth',
 
   components: {
-    ActionMenu,
-    BaseDialog,
-    ConfirmForm,
     DocumentList,
-    DocumentUploadForm,
-    DocumentExportForm
-  },
-
-  data() {
-    return {
-      importDialog: false,
-      exportDialog: false,
-      deleteDialog: false,
-      menuItems: [
-        { title: 'Import', icon: 'backup', event: 'upload' },
-        { title: 'Export', icon: 'archive', event: 'download' }
-      ]
-    }
-  },
-
-  computed: {
-    ...mapGetters('documents', ['formatList', 'isDocumentSelected']),
-    ...mapState('documents', ['selected'])
-  },
-
-  methods: {
-    ...mapActions('documents', ['uploadDocument', 'exportDocument', 'deleteDocument'])
+    DocumentActionMenu,
+    DocumentDeletionButton
   },
 
   validate({ params }) {