diff --git a/src/Plugin/Field/FieldFormatter/ImageFieldTokensFormatter.php b/src/Plugin/Field/FieldFormatter/ImageFieldTokensFormatter.php
index f0bfdc9..449d4bf 100644
--- a/src/Plugin/Field/FieldFormatter/ImageFieldTokensFormatter.php
+++ b/src/Plugin/Field/FieldFormatter/ImageFieldTokensFormatter.php
@@ -172,11 +172,13 @@ class ImageFieldTokensFormatter extends ImageFormatter {
           }
         }
       }
-      /* @var object $entity */
-      $entity_type = $entity->getEntityTypeId();
+      $data = [];
+      if ($entity) {
+        $data[$entity->getEntityTypeId()] = $entity;
+      }
       // Replace entity tokens.
-      $alt_token = $this->tokenService->replace($item_values['alt'], [$entity_type => $entity]);
-      $title_token = $this->tokenService->replace($item_values['title'], [$entity_type => $entity]);
+      $alt_token = $this->tokenService->replace($item_values['alt'], $data);
+      $title_token = $this->tokenService->replace($item_values['title'], $data);
       // Set converted values to the item.
       $item_values['alt'] = $alt_token;
       $item_values['title'] = $title_token;