diff --git a/tests/phpunit/tests/widgets/custom-html-widget.php b/tests/phpunit/tests/widgets/custom-html-widget.php
index 916cc1fc46..dcd03d2007 100644
--- a/tests/phpunit/tests/widgets/custom-html-widget.php
+++ b/tests/phpunit/tests/widgets/custom-html-widget.php
@@ -71,7 +71,7 @@ class Test_WP_Widget_Custom_HTML extends WP_UnitTestCase {
 		$this->assertNotEmpty( $this->widget_text_args );
 		$this->assertContains( '[filter:widget_text][filter:widget_custom_html_content]', $output );
 		$this->assertContains( '<section id="custom_html-5" class="widget_text widget widget_custom_html">', $output );
-		$this->assertContains( 'class="widget_text widget widget_custom_html"', $output );
+		$this->assertContains( '<div class="textwidget custom-html-widget">', $output );
 		$this->assertNotContains( '<p>', $output );
 		$this->assertNotContains( '<br>', $output );
 		$this->assertNotContains( '</u>', $output );