diff --git a/src/Tests/EntityEmbedFilterTest.php b/src/Tests/EntityEmbedFilterTest.php
index d46a27d..b18c3d3 100644
--- a/src/Tests/EntityEmbedFilterTest.php
+++ b/src/Tests/EntityEmbedFilterTest.php
@@ -15,6 +15,13 @@ namespace Drupal\entity_embed\Tests;
 class EntityEmbedFilterTest extends EntityEmbedTestBase {
 
   /**
+   * Modules to enable.
+   *
+   * @var array
+   */
+  public static $modules = ['file', 'image', 'entity_embed', 'entity_embed_test', 'node', 'ckeditor'];
+
+  /**
    * Tests the entity_embed filter.
    *
    * Ensures that entities are getting rendered when correct data attributes
@@ -118,6 +125,24 @@ class EntityEmbedFilterTest extends EntityEmbedTestBase {
     $this->drupalget('node/' . $node->id());
     $this->assertNoText($this->node->body->value, 'embedded node exists in page');
     $this->assertRaw('<div data-entity-type="node" data-entity-id');
+
+    // Test that attributes are correctly added when image formatter is used.
+    /** @var \Drupal\file\FileInterface $image */
+    $image = $this->getTestFile('image');
+    $image->setPermanent();
+    $image->save();
+    $content = '<drupal-entity data-entity-type="file" data-entity-uuid="' . $image->uuid() . '" data-entity-embed-display="image:image" data-entity-embed-settings=\'{"image_style":"","image_link":""}\' data-align="left" data-caption="test caption" alt="This is alt text" title="This is title text">This placeholder should not be rendered.</drupal-entity>';
+    $settings = [];
+    $settings['type'] = 'page';
+    $settings['title'] = 'test entity image formatter';
+    $settings['body'] = [['value' => $content, 'format' => 'custom_format']];
+    $node = $this->drupalCreateNode($settings);
+    $this->drupalget('node/' . $node->id());
+    $this->assertRaw('<img src', 'IMG tag found.');
+    $this->assertRaw('data-caption="test caption"', 'Caption found.');
+    $this->assertRaw('data-align="left"', 'Alignment information found.');
+    $this->assertTrue((bool) $this->xpath("//img[@alt='This is alt text']"), 'Alt text found');
+    $this->assertTrue((bool) $this->xpath("//img[@title='This is title text']"), 'Title text found');
   }
 
 }