diff --git a/src/Plugin/Field/FieldFormatter/VidstackPlayerVideoFormatter.php b/src/Plugin/Field/FieldFormatter/VidstackPlayerVideoFormatter.php
index 5608ac88a56e6d34ea31dc2553f4dca78c391887..69f9257b5f87c3fe5ffb99978c146ffe9f8aefd6 100644
--- a/src/Plugin/Field/FieldFormatter/VidstackPlayerVideoFormatter.php
+++ b/src/Plugin/Field/FieldFormatter/VidstackPlayerVideoFormatter.php
@@ -236,7 +236,7 @@ class VidstackPlayerVideoFormatter extends FileMediaFormatterBase implements Fil
    */
   public function getEntityImageFields(): array {
     $imageFieldsArray = [];
-    $imageFields = array_filter($this->entityFieldManager->getFieldDefinitions('media', 'video'),
+    $imageFields = array_filter($this->entityFieldManager->getFieldDefinitions($this->fieldDefinition->getTargetEntityTypeId(), $this->fieldDefinition->getTargetBundle()),
     function (FieldDefinitionInterface $field) {
       return $field->getType() === 'image';
     });