diff --git a/modules/media/js/plugins/drupalmedia/plugin.es6.js b/modules/media/js/plugins/drupalmedia/plugin.es6.js
index 3334b6fae7..fc2588ab07 100644
--- a/core/modules/media/js/plugins/drupalmedia/plugin.es6.js
+++ b/core/modules/media/js/plugins/drupalmedia/plugin.es6.js
@@ -337,7 +337,7 @@
           const editButton = CKEDITOR.dom.element.createFromHtml(
             Drupal.theme('mediaEmbedEditButton'),
           );
-          embeddedMedia.getFirst().insertBeforeMe(editButton);
+          embeddedMedia.insertBeforeMe(editButton);
 
           // Make the edit button do things.
           const widget = this;
diff --git a/modules/media/js/plugins/drupalmedia/plugin.js b/modules/media/js/plugins/drupalmedia/plugin.js
index c26512e732..dd2f1714e4 100644
--- a/core/modules/media/js/plugins/drupalmedia/plugin.js
+++ b/core/modules/media/js/plugins/drupalmedia/plugin.js
@@ -237,7 +237,7 @@
 
           embeddedMedia.setStyle('position', 'relative');
           var editButton = CKEDITOR.dom.element.createFromHtml(Drupal.theme('mediaEmbedEditButton'));
-          embeddedMedia.getFirst().insertBeforeMe(editButton);
+          embeddedMedia.insertBeforeMe(editButton);
           var widget = this;
           this.element.findOne('.media-library-item__edit').on('click', function (event) {
             var saveCallback = function saveCallback(values) {