diff --git a/core/modules/ckeditor5/tests/src/FunctionalJavascript/ImageTestProviderTrait.php b/core/modules/ckeditor5/tests/src/FunctionalJavascript/ImageTestProviderTrait.php
index fa6652781540f7f4c4237194d59505ab884fcb7f..6e6daf548c98a094c805cc01f6d8ac1130479dc6 100644
--- a/core/modules/ckeditor5/tests/src/FunctionalJavascript/ImageTestProviderTrait.php
+++ b/core/modules/ckeditor5/tests/src/FunctionalJavascript/ImageTestProviderTrait.php
@@ -281,8 +281,8 @@ public function testResize(bool $is_resize_enabled): void {
     $this->drupalGet('node/add');
     $page->fillField('title[0][value]', 'My test content');
     $this->addImage();
-    $image_figure = $assert_session->waitForElementVisible('css', 'figure');
-    $this->assertSame($is_resize_enabled, $image_figure->hasClass('ck-widget_with-resizer'));
+    $selector = $is_resize_enabled ? 'figure.ck-widget_with-resizer' : 'figure:not(.ck-widget_with-resizer)';
+    $this->assertNotEmpty($assert_session->waitForElementVisible('css', $selector));
   }
 
   /**