diff --git a/tests/src/Unit/Plugin/processor/TranslateEntityProcessorTest.php b/tests/src/Unit/Plugin/processor/TranslateEntityProcessorTest.php
index 91a0f1c..9e825bf 100644
--- a/tests/src/Unit/Plugin/processor/TranslateEntityProcessorTest.php
+++ b/tests/src/Unit/Plugin/processor/TranslateEntityProcessorTest.php
@@ -10,7 +10,6 @@
 use Drupal\Core\TypedData\ComplexDataDefinitionInterface;
 use Drupal\Core\TypedData\DataReferenceDefinitionInterface;
 use Drupal\facets\Entity\Facet;
-use Drupal\facets\Plugin\facets\facet_source\SearchApiDisplay;
 use Drupal\facets\Plugin\facets\processor\TranslateEntityProcessor;
 use Drupal\facets\Result\Result;
 use Drupal\node\Entity\Node;
@@ -79,23 +78,14 @@ protected function setUp() {
       ->method('getPropertyDefinitions')
       ->willReturn([$property_definition]);
 
-    // Create a search api based facet source and link the index to it.
-    $facet_source = $this->getMockBuilder(SearchApiDisplay::class)
-      ->disableOriginalConstructor()
-      ->getMock();
-    $facet_source->expects($this->any())
-      ->method('getDataDefinition')
-      ->willReturn($data_definition);
-
     // Create the actual facet.
     $this->facet = $this->getMockBuilder(Facet::class)
       ->disableOriginalConstructor()
       ->getMock();
-
-    // Return the configured facet source.
     $this->facet->expects($this->any())
-      ->method('getFacetSource')
-      ->willReturn($facet_source);
+      ->method('getDataDefinition')
+      ->willReturn($data_definition);
+
     // Add a field identifier.
     $this->facet->expects($this->any())
       ->method('getFieldIdentifier')