diff --git a/src/Resolver/MediaProviderResolver.php b/src/Resolver/MediaProviderResolver.php
index 157b62549..5d640cef7 100755
--- a/src/Resolver/MediaProviderResolver.php
+++ b/src/Resolver/MediaProviderResolver.php
@@ -29,7 +29,7 @@ public function __construct(ServiceRegistryInterface $providerRegistry)
     public function resolveProvider(MediaInterface $media): ProviderInterface
     {
         /** @var ProviderInterface $provider */
-        $provider = $this->providerRegistry->get($media->getFileType());
+        $provider = $this->providerRegistry->get($media->getType());
 
         return $provider;
     }
diff --git a/tests/Behat/Context/Setup/MediaContext.php b/tests/Behat/Context/Setup/MediaContext.php
index aef99394d..5366d3239 100755
--- a/tests/Behat/Context/Setup/MediaContext.php
+++ b/tests/Behat/Context/Setup/MediaContext.php
@@ -127,7 +127,7 @@ private function createMedia(
         $media->setCurrentLocale('en_US');
         $media->setName($name);
         $media->setDescription($description);
-        $media->setFileType($fileType);
+        $media->setType($fileType);
 
         return $media;
     }