diff --git a/core/modules/ckeditor5/tests/src/FunctionalJavascript/ImageTestProviderTrait.php b/core/modules/ckeditor5/tests/src/FunctionalJavascript/ImageTestProviderTrait.php
index 9e7571bc79076da27c66c760eb37cb83e456a8a0..2588cce74b07c64f35db1d62b4c146a3cc754be3 100644
--- a/core/modules/ckeditor5/tests/src/FunctionalJavascript/ImageTestProviderTrait.php
+++ b/core/modules/ckeditor5/tests/src/FunctionalJavascript/ImageTestProviderTrait.php
@@ -289,8 +289,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));
   }
 
   /**