tests/src/Kernel/EntityEmbedFilterOverridesTest.php | 9 +++------
 tests/src/Kernel/EntityEmbedFilterTest.php          | 9 ++++-----
 2 files changed, 7 insertions(+), 11 deletions(-)

diff --git a/tests/src/Kernel/EntityEmbedFilterOverridesTest.php b/tests/src/Kernel/EntityEmbedFilterOverridesTest.php
index 7f150f8..1e27bfe 100644
--- a/tests/src/Kernel/EntityEmbedFilterOverridesTest.php
+++ b/tests/src/Kernel/EntityEmbedFilterOverridesTest.php
@@ -65,8 +65,6 @@ class EntityEmbedFilterOverridesTest extends EntityEmbedFilterTestBase {
       'data-entity-uuid' => $this->image->uuid(),
       'data-entity-embed-display' => 'image:image',
       'data-entity-embed-display-settings' => '{"image_style":"","image_link":""}',
-      'data-align' => 'left',
-      'data-caption' => 'test caption',
       'alt' => 'This is alt text',
       'title' => 'This is title text',
     ]);
@@ -78,11 +76,10 @@ class EntityEmbedFilterOverridesTest extends EntityEmbedFilterTestBase {
 
     $dom = Html::load($output);
     $xpath = new \DOMXPath($dom);
-    /** @var \DOMNode $rendered_embed */
-    $rendered_embed = $xpath->query('//div[contains(@class, "embedded-entity")]')[0];
-    $this->assertHasAttributes($rendered_embed, [
+    /** @var \DOMNode $embedded_entity_container */
+    $embedded_entity_container = $xpath->query('//div[contains(@class, "embedded-entity")]')[0];
+    $this->assertHasAttributes($embedded_entity_container, [
       'alt' => 'This is alt text',
-      'data-align' => 'left',
       'data-entity-embed-display' => 'image:image',
       'data-entity-type' => 'file',
       'data-entity-uuid' => $this->image->uuid(),
diff --git a/tests/src/Kernel/EntityEmbedFilterTest.php b/tests/src/Kernel/EntityEmbedFilterTest.php
index 4627301..4b6ba9a 100644
--- a/tests/src/Kernel/EntityEmbedFilterTest.php
+++ b/tests/src/Kernel/EntityEmbedFilterTest.php
@@ -57,11 +57,11 @@ class EntityEmbedFilterTest extends EntityEmbedFilterTestBase {
 
     $dom = Html::load($output);
     $xpath = new \DOMXPath($dom);
-    $rendered_embed = $xpath->query('//div[contains(@class, "embedded-entity")]')[0];
-    $this->assertNotEmpty($rendered_embed);
-    $this->assertHasAttributes($rendered_embed, $expected_attributes);
+    $embedded_entity_container = $xpath->query('//div[contains(@class, "embedded-entity")]')[0];
+    $this->assertNotEmpty($embedded_entity_container);
+    $this->assertHasAttributes($embedded_entity_container, $expected_attributes);
     if (empty($embed_attributes['data-entity-id'])) {
-      $this->assertFalse($rendered_embed->hasAttribute('data-entity-id'));
+      $this->assertFalse($embedded_entity_container->hasAttribute('data-entity-id'));
     }
   }
 
@@ -366,7 +366,6 @@ class EntityEmbedFilterTest extends EntityEmbedFilterTestBase {
 
     $dom = Html::load($output);
     $xpath = new \DOMXPath($dom);
-    /** @var \DOMNode $rendered_embed */
     $this->assertSame(0, $xpath->query('//div[contains(@class, "embedded-entity")]/a')->length);
     $this->assertSame($this->node->label(), $xpath->query('//div[contains(@class, "embedded-entity")]')[0]->textContent);
   }