diff --git a/core/modules/media_library/tests/src/FunctionalJavascript/MediaLibraryTest.php b/core/modules/media_library/tests/src/FunctionalJavascript/MediaLibraryTest.php
index f2a4afafbe..caa222fbe6 100644
--- a/core/modules/media_library/tests/src/FunctionalJavascript/MediaLibraryTest.php
+++ b/core/modules/media_library/tests/src/FunctionalJavascript/MediaLibraryTest.php
@@ -672,7 +672,7 @@ public function testWidgetUpload() {
     $assert_session->assertWaitOnAjaxRequest();
     $this->assertJsCondition('jQuery("#media-library-add-form-wrapper .media-library-add-form-media").is(":focus")');
     $assert_session->pageTextContains('The media item has been created. Fill in any required fields and save to add it to the media library.');
-    $this->assertSame('Added media items', $assert_session->elementExists('css', '.media-library-add-form-media')->getAttribute('aria-label'));
+    $assert_session->elementAttributeContains('css', '.media-library-add-form-media', 'aria-label', 'Added media items');
     $assert_session->elementExists('css', '.media-library-add-form--with-input');
     $assert_session->elementNotExists('css', '.media-library-add-form--without-input');
     // We do not have a pre-selected items, so the container should not be added
@@ -758,9 +758,8 @@ public function testWidgetUpload() {
     $png_uri_3 = $file_system->copy($png_image->uri);
     $page->attachFileToField('Add files', $this->container->get('file_system')->realpath($png_uri_3));
     $assert_session->assertWaitOnAjaxRequest();
-    $selection_area = $assert_session->elementExists('css', '.media-library-add-form-selection');
-    $this->assertSame('Selected media items', $selection_area->getAttribute('aria-label'));
-    $assert_session->checkboxChecked("Select $existing_media_name", $selection_area);
+    $assert_session->elementAttributeContains('css', '.media-library-add-form-selection', 'aria-label', 'Selected media items');
+    $assert_session->checkboxChecked("Select $existing_media_name");
     $page->fillField('Name', 'Unlimited Cardinality Image');
     $page->fillField('Alternative text', $this->randomString());
     $assert_session->elementExists('css', '.ui-dialog-buttonpane')->pressButton('Save and select');