diff --git a/src/EntityEmbedBuilder.php b/src/EntityEmbedBuilder.php
index cd4d235..2148388 100644
--- a/src/EntityEmbedBuilder.php
+++ b/src/EntityEmbedBuilder.php
@@ -76,7 +76,7 @@ class EntityEmbedBuilder implements EntityEmbedBuilderInterface {
     if (!empty($context['data-embed-button']) &&
         ($embed_button = EmbedButton::load($context['data-embed-button'])) &&
         ($preview_display_plugin = $embed_button->getTypeSetting('preview_display_plugin', '')) &&
-        (\Drupal::routeMatch()->getRouteName() === 'embed.preview')) {
+        (\Drupal::routeMatch()->getRouteName() === 'entity_embed.preview')) {
       $context['data-entity-embed-display'] = $preview_display_plugin;
     }