diff --git a/class-block-unit-test.php b/class-block-unit-test.php
index 9269bfc..5d01bac 100644
--- a/class-block-unit-test.php
+++ b/class-block-unit-test.php
@@ -210,6 +210,9 @@ public function upgrade_completed( $upgrader_object, $options ) {
 	 */
 	public function content() {
 
+		$layout_settings = wp_get_global_settings(array( 'layout' ));
+		$content_size_in_theme_json = $layout_settings['contentSize'];
+
 		$content = '';
 
 		$content .= '
@@ -598,7 +601,7 @@ public function content() {
 			<!-- /wp:paragraph -->
 		';
 
-		if ( get_theme_support( 'align-wide' ) ) {
+		if ( get_theme_support( 'align-wide' )  || isset( $content_size_in_theme_json ) ) {
 			$content .= '
 				<!-- wp:pullquote {"align":"wide"} -->
 				<blockquote class="wp-block-pullquote alignwide">
@@ -666,7 +669,7 @@ public function content() {
 			<!-- /wp:image -->
 		';
 
-		if ( get_theme_support( 'align-wide' ) ) {
+		if ( get_theme_support( 'align-wide' )  || isset(  $content_size_in_theme_json )  ) {
 			$content .= '
 				<!-- wp:heading {"level":3} -->
 				<h3>' . esc_html__( 'Wide aligned', '@@textdomain' ) . '</h3>
@@ -724,7 +727,7 @@ public function content() {
 			<!-- /wp:paragraph -->
 		';
 
-		if ( get_theme_support( 'align-wide' ) ) {
+		if ( get_theme_support( 'align-wide' )  || isset( $content_size_in_theme_json ) ) {
 			$content .= '
 				<!-- wp:heading {"level":3} -->
 				<h3>' . esc_html__( 'Wide aligned', '@@textdomain' ) . '</h3>
@@ -760,7 +763,7 @@ public function content() {
 
 		';
 
-		if ( get_theme_support( 'align-wide' ) ) {
+		if ( get_theme_support( 'align-wide' )  || isset( $content_size_in_theme_json ) ) {
 			$content .= '
 				<!-- wp:heading {"level":3} -->
 				<h3>' . esc_html__( 'Wide aligned', '@@textdomain' ) . '</h3>
@@ -1037,7 +1040,7 @@ public function content() {
 			<!-- /wp:gallery -->
 		';
 
-		if ( get_theme_support( 'align-wide' ) ) {
+		if ( get_theme_support( 'align-wide' )  || isset(  $content_size_in_theme_json ) ) {
 			$content .= '
 				<!-- wp:heading {"level":2} -->
 				<h2>Wide aligned Gallery Blocks</h2>