diff --git a/config/install/core.entity_form_display.media.document.default.yml b/config/install/core.entity_form_display.media.document.default.yml
deleted file mode 100644
index ecb0027..0000000
--- a/config/install/core.entity_form_display.media.document.default.yml
+++ /dev/null
@@ -1,56 +0,0 @@
-uuid: 3bd769cd-c261-4b77-92e4-e1fa86082218
-langcode: en
-status: true
-dependencies:
-  config:
-    - field.field.media.document.field_file_size
-    - field.field.media.document.field_image_type
-    - field.field.media.document.field_media_file
-    - media_entity.bundle.document
-  module:
-    - file
-id: media.document.default
-targetEntityType: media
-bundle: document
-mode: default
-content:
-  created:
-    type: datetime_timestamp
-    weight: 3
-    settings: {  }
-    third_party_settings: {  }
-  field_file_size:
-    weight: 5
-    settings:
-      placeholder: ''
-    third_party_settings: {  }
-    type: number
-  field_image_type:
-    weight: 4
-    settings:
-      size: 30
-      placeholder: ''
-    third_party_settings: {  }
-    type: string_textfield
-  field_media_file:
-    weight: 1
-    settings:
-      progress_indicator: throbber
-    third_party_settings: {  }
-    type: file_generic
-  name:
-    type: string_textfield
-    weight: 0
-    settings:
-      size: 60
-      placeholder: ''
-    third_party_settings: {  }
-  uid:
-    type: entity_reference_autocomplete
-    weight: 2
-    settings:
-      match_operator: CONTAINS
-      size: 60
-      placeholder: ''
-    third_party_settings: {  }
-hidden: {  }
diff --git a/config/install/core.entity_form_display.media.image.default.yml b/config/install/core.entity_form_display.media.image.default.yml
deleted file mode 100644
index d880efc..0000000
--- a/config/install/core.entity_form_display.media.image.default.yml
+++ /dev/null
@@ -1,67 +0,0 @@
-uuid: e17a0c90-5021-459d-bf2f-4e5ee2d65373
-langcode: en
-status: true
-dependencies:
-  config:
-    - field.field.media.image.field_image_height
-    - field.field.media.image.field_image_type
-    - field.field.media.image.field_image_width
-    - field.field.media.image.field_media_file
-    - media_entity.bundle.image
-  module:
-    - file
-_core:
-  default_config_hash: dJTJ3jGpyJ_xKi3pB1RHTjUPb2ErsDJtf2SbyWPTGjM
-id: media.image.default
-targetEntityType: media
-bundle: image
-mode: default
-content:
-  created:
-    type: datetime_timestamp
-    weight: 3
-    settings: {  }
-    third_party_settings: {  }
-  field_image_height:
-    weight: 6
-    settings:
-      size: 10
-      placeholder: ''
-    third_party_settings: {  }
-    type: string_textfield
-  field_image_type:
-    weight: 4
-    settings:
-      size: 30
-      placeholder: ''
-    third_party_settings: {  }
-    type: string_textfield
-  field_image_width:
-    weight: 5
-    settings:
-      size: 10
-      placeholder: ''
-    third_party_settings: {  }
-    type: string_textfield
-  field_media_file:
-    weight: 1
-    settings:
-      progress_indicator: throbber
-    third_party_settings: {  }
-    type: file_generic
-  name:
-    type: string_textfield
-    weight: 0
-    settings:
-      size: 60
-      placeholder: ''
-    third_party_settings: {  }
-  uid:
-    type: entity_reference_autocomplete
-    weight: 2
-    settings:
-      match_operator: CONTAINS
-      size: 60
-      placeholder: ''
-    third_party_settings: {  }
-hidden: {  }
diff --git a/config/install/core.entity_view_display.media.document.default.yml b/config/install/core.entity_view_display.media.document.default.yml
deleted file mode 100644
index 98dde11..0000000
--- a/config/install/core.entity_view_display.media.document.default.yml
+++ /dev/null
@@ -1,71 +0,0 @@
-uuid: 29274fe5-0266-4960-bbac-c076766bbb1e
-langcode: en
-status: true
-dependencies:
-  config:
-    - field.field.media.document.field_file_size
-    - field.field.media.document.field_image_type
-    - field.field.media.document.field_media_file
-    - image.style.thumbnail
-    - media_entity.bundle.document
-  module:
-    - file
-    - image
-    - user
-id: media.document.default
-targetEntityType: media
-bundle: document
-mode: default
-content:
-  created:
-    type: timestamp
-    weight: 1
-    label: hidden
-    settings:
-      date_format: medium
-      custom_date_format: ''
-      timezone: ''
-    third_party_settings: {  }
-  field_file_size:
-    weight: 6
-    label: inline
-    settings:
-      thousand_separator: ''
-      prefix_suffix: true
-    third_party_settings: {  }
-    type: number_integer
-  field_image_type:
-    weight: 5
-    label: inline
-    settings:
-      link_to_entity: false
-    third_party_settings: {  }
-    type: string
-  field_media_file:
-    weight: 4
-    label: above
-    settings: {  }
-    third_party_settings: {  }
-    type: file_default
-  name:
-    label: hidden
-    type: string
-    weight: 0
-    settings:
-      link_to_entity: false
-    third_party_settings: {  }
-  thumbnail:
-    type: image
-    weight: 3
-    label: hidden
-    settings:
-      image_style: thumbnail
-      image_link: ''
-    third_party_settings: {  }
-  uid:
-    type: author
-    weight: 2
-    label: hidden
-    settings: {  }
-    third_party_settings: {  }
-hidden: {  }
diff --git a/config/install/core.entity_view_display.media.image.default.yml b/config/install/core.entity_view_display.media.image.default.yml
deleted file mode 100644
index febb0d3..0000000
--- a/config/install/core.entity_view_display.media.image.default.yml
+++ /dev/null
@@ -1,80 +0,0 @@
-uuid: 6b78fa61-1f2b-4371-81b5-c6c39f12dbac
-langcode: en
-status: true
-dependencies:
-  config:
-    - field.field.media.image.field_image_height
-    - field.field.media.image.field_image_type
-    - field.field.media.image.field_image_width
-    - field.field.media.image.field_media_file
-    - image.style.thumbnail
-    - media_entity.bundle.image
-  module:
-    - file
-    - image
-    - user
-_core:
-  default_config_hash: n8PAik9P8cgaVsY5MEhTE8yIp11iWvGPM_cHNEpJU_g
-id: media.image.default
-targetEntityType: media
-bundle: image
-mode: default
-content:
-  created:
-    label: hidden
-    type: timestamp
-    weight: 1
-    settings:
-      date_format: medium
-      custom_date_format: ''
-      timezone: ''
-    third_party_settings: {  }
-  field_image_height:
-    weight: 7
-    label: inline
-    settings:
-      link_to_entity: false
-    third_party_settings: {  }
-    type: string
-  field_image_type:
-    weight: 5
-    label: inline
-    settings:
-      link_to_entity: false
-    third_party_settings: {  }
-    type: string
-  field_image_width:
-    weight: 6
-    label: inline
-    settings:
-      link_to_entity: false
-    third_party_settings: {  }
-    type: string
-  field_media_file:
-    weight: 4
-    label: above
-    settings: {  }
-    third_party_settings: {  }
-    type: file_default
-  name:
-    label: hidden
-    type: string
-    weight: 0
-    settings:
-      link_to_entity: false
-    third_party_settings: {  }
-  thumbnail:
-    type: image
-    weight: 3
-    label: hidden
-    settings:
-      image_style: thumbnail
-      image_link: ''
-    third_party_settings: {  }
-  uid:
-    label: hidden
-    type: author
-    weight: 2
-    settings: {  }
-    third_party_settings: {  }
-hidden: {  }
diff --git a/config/install/media_entity.bundle.document.yml b/config/install/media_entity.bundle.document.yml
deleted file mode 100644
index d5e564d..0000000
--- a/config/install/media_entity.bundle.document.yml
+++ /dev/null
@@ -1,17 +0,0 @@
-uuid: 1f5295b0-acab-443c-995d-1d6f822bf19d
-langcode: en
-status: true
-dependencies:
-  module:
-    - media_entity_document
-id: document
-label: Document
-description: 'A product of word processing or other office/productivity software, usually text-heavy; examples include spreadsheets and text documents.'
-type: document
-queue_thumbnail_downloads: false
-new_revision: false
-type_configuration:
-  source_field: field_media_file
-field_map:
-  mime: field_image_type
-  size: field_file_size
diff --git a/config/install/media_entity.bundle.image.yml b/config/install/media_entity.bundle.image.yml
deleted file mode 100644
index 54b2fdd..0000000
--- a/config/install/media_entity.bundle.image.yml
+++ /dev/null
@@ -1,21 +0,0 @@
-uuid: fd185b00-88ab-4b07-9887-fc671220ea20
-langcode: en
-status: true
-dependencies:
-  module:
-    - media_entity_image
-_core:
-  default_config_hash: RkcQ4hOmuAUscKYIVXXPl7s-WVCLwpD_IeouMovMWI0
-id: image
-label: Image
-description: 'A graphic in a binary format; examples include GIFs and PNGs.'
-type: image
-queue_thumbnail_downloads: false
-new_revision: false
-type_configuration:
-  source_field: field_media_file
-  gather_exif: false
-field_map:
-  mime: field_image_type
-  width: field_image_width
-  height: field_image_height
diff --git a/src/FileMigrateEventSubscriber.php b/src/FileMigrateEventSubscriber.php
index 6d56e06..e963041 100644
--- a/src/FileMigrateEventSubscriber.php
+++ b/src/FileMigrateEventSubscriber.php
@@ -43,7 +43,7 @@ class FileMigrateEventSubscriber implements EventSubscriberInterface {
    *   The event triggered.
    */
   public function onPostRowSave(MigratePostRowSaveEvent $event) {
-    if ($event->getMigration()->getPluginId() != 'd6_file') {
+    if ($event->getMigration()->getPluginId() != 'files_migration23') {
       return;
     }
 
@@ -61,7 +61,7 @@ class FileMigrateEventSubscriber implements EventSubscriberInterface {
       'bundle' => $bundle,
       'uid' => $file->uid,
       'name' => $file->name,
-      'field_media_file' => [
+      'field_document' => [
         'target_id' => $destinationIds[0]
       ],
     ]);