diff --git a/src/Hook/TokenTokensHooks.php b/src/Hook/TokenTokensHooks.php
index f9cec55..02b52f3 100644
--- a/src/Hook/TokenTokensHooks.php
+++ b/src/Hook/TokenTokensHooks.php
@@ -7,6 +7,7 @@ use Drupal\Core\Config\ConfigFactoryInterface;
 use Drupal\Core\Controller\TitleResolverInterface;
 use Drupal\Core\Datetime\DateFormatterInterface;
 use Drupal\Core\Entity\ContentEntityInterface;
+use Drupal\Core\Entity\EntityInterface;
 use Drupal\Core\Entity\EntityRepositoryInterface;
 use Drupal\Core\Entity\EntityTypeManagerInterface;
 use Drupal\Core\Entity\FieldableEntityInterface;
@@ -1181,7 +1182,7 @@ final class TokenTokensHooks {
           continue;
         }
 
-        if (isset($field_item->$property_name) && ($field_item->$property_name instanceof FieldableEntityInterface)) {
+        if (isset($field_item->$property_name) && ($field_item->$property_name instanceof EntityInterface)) {
           // Entity reference field.
           $entity = $field_item->$property_name;
           // Obtain the referenced entity with the correct language.