diff --git a/modules/autotagger_search_in_text/src/Plugin/Autotagger/SearchInTextFields.php b/modules/autotagger_search_in_text/src/Plugin/Autotagger/SearchInTextFields.php
index 46d76a6..6c127cd 100644
--- a/modules/autotagger_search_in_text/src/Plugin/Autotagger/SearchInTextFields.php
+++ b/modules/autotagger_search_in_text/src/Plugin/Autotagger/SearchInTextFields.php
@@ -343,12 +343,12 @@ class SearchInTextFields extends AutotaggerPluginBase {
   /**
    * Get the values from all text fields on the entity.
    *
-   * @param \Drupal\Core\Entity\ContentEntityInterface $entity
+   * @param \Drupal\node\Entity\Node $entity
    *   Entity from an entity reference field.
    * @param array $text_array
    *   Array that collects the values from the fields.
    */
-  protected function processReferencedEntity(ContentEntityInterface $entity, array &$text_array) {
+  protected function processReferencedEntity(Node $entity, array &$text_array) {
     $fields = $entity->getFields();
     foreach ($fields as $name => $field) {
       $type = $entity->getFieldDefinition($name)->getType();