diff --git a/src/Plugin/Action/LinkCaption.php b/src/Plugin/Action/LinkCaption.php
index 18a385c..6228e7c 100644
--- a/src/Plugin/Action/LinkCaption.php
+++ b/src/Plugin/Action/LinkCaption.php
@@ -82,9 +82,9 @@ final class LinkCaption extends ActionBase implements ContainerFactoryPluginInte
    * {@inheritdoc}
    */
   public function execute(ContentEntityInterface $entity = NULL): void {
-    $transcript_media = $this->getMediaByUri('http://pcdm.org/use#Transcript');
-    $host_media = $this->getMediaByUri('http://pcdm.org/use#ServiceFile')
-        ?? $this->getMediaByUri('http://pcdm.org/use#OriginalFile');
+    $transcript_media = $this->getMediaByUri($entity, 'http://pcdm.org/use#Transcript');
+    $host_media = $this->getMediaByUri($entity, 'http://pcdm.org/use#ServiceFile')
+        ?? $this->getMediaByUri($entity, 'http://pcdm.org/use#OriginalFile');
 
     if ($transcript_media && $host_media
         && $transcript_media->bundle() === 'able_player_caption'