tests/src/Kernel/EntityEmbedFilterOverridesTest.php | 1 -
 tests/src/Kernel/EntityEmbedFilterTest.php          | 1 -
 tests/src/Kernel/EntityEmbedFilterTestBase.php      | 1 +
 3 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/tests/src/Kernel/EntityEmbedFilterOverridesTest.php b/tests/src/Kernel/EntityEmbedFilterOverridesTest.php
index d44a20b..7f150f8 100644
--- a/tests/src/Kernel/EntityEmbedFilterOverridesTest.php
+++ b/tests/src/Kernel/EntityEmbedFilterOverridesTest.php
@@ -145,7 +145,6 @@ class EntityEmbedFilterOverridesTest extends EntityEmbedFilterTestBase {
     /** @var \Drupal\filter\FilterProcessResult $filter_result */
     $filter_result = $this->filter->process($input, 'en');
     $output = $filter_result->getProcessedText();
-    $this->assertNotContains('drupal-entity data-entity-type="media" data-entity', $output);
     $this->assertEntityEmbedFilterHasRun($output);
     $dom = Html::load($output);
     $xpath = new \DOMXPath($dom);
diff --git a/tests/src/Kernel/EntityEmbedFilterTest.php b/tests/src/Kernel/EntityEmbedFilterTest.php
index 4da5bec..4627301 100644
--- a/tests/src/Kernel/EntityEmbedFilterTest.php
+++ b/tests/src/Kernel/EntityEmbedFilterTest.php
@@ -50,7 +50,6 @@ class EntityEmbedFilterTest extends EntityEmbedFilterTestBase {
       $this->assertEmpty($output);
       return;
     }
-    $this->assertNotContains('drupal-entity data-entity-type="node" data-entity', $output);
     $this->assertEntityEmbedFilterHasRun($output);
 
     $this->assertContainsMultiple($contains, $output);
diff --git a/tests/src/Kernel/EntityEmbedFilterTestBase.php b/tests/src/Kernel/EntityEmbedFilterTestBase.php
index 7a33cf9..16d3568 100644
--- a/tests/src/Kernel/EntityEmbedFilterTestBase.php
+++ b/tests/src/Kernel/EntityEmbedFilterTestBase.php
@@ -135,6 +135,7 @@ abstract class EntityEmbedFilterTestBase extends KernelTestBase {
    * @see \Drupal\Tests\entity_embed\Kernel\EntityEmbedFilterTestBase::createEmbedCode()
    */
   protected function assertEntityEmbedFilterHasRun($output) {
+    $this->assertNotContains('<drupal-entity', $output);
     $this->assertNotContains('This placeholder should not be rendered.', $output);
   }