diff --git a/test/generator/samples/render-in-iframe/Widget.html b/test/generator/samples/render-in-iframe/Widget.html
new file mode 100644
index 0000000000..1d893b8bcd
--- /dev/null
+++ b/test/generator/samples/render-in-iframe/Widget.html
@@ -0,0 +1,7 @@
+<p>test</p>
+
+<style>
+	p {
+		color: red;
+	}
+</style>
diff --git a/test/generator/samples/render-in-iframe/_config.js b/test/generator/samples/render-in-iframe/_config.js
index 190a2b57de..25f3a6bf54 100644
--- a/test/generator/samples/render-in-iframe/_config.js
+++ b/test/generator/samples/render-in-iframe/_config.js
@@ -1,21 +1,19 @@
 export default {
 	test ( assert, component, target, window ) {
+		const testStyles = ([control, test]) => {
+			assert.equal( window.getComputedStyle( control ).color, 'blue' );
+			assert.equal( window.getComputedStyle( test ).color, 'red' );
+		};
+
 		const iframe = window.document.createElement('iframe');
 		window.document.body.appendChild(iframe);
-
 		const otherTarget = iframe.contentWindow.document.body;
 
 		new component.constructor({
 			target: otherTarget
 		});
 
-		assert.equal(
-			window.getComputedStyle(target.querySelector('h1')).color,
-			'rgb(255, 0, 0)'
-		);
-		assert.equal(
-			window.getComputedStyle(otherTarget.querySelector('h1')).color,
-			'rgb(255, 0, 0)'
-		);
+		testStyles(target.querySelectorAll( 'p' ));
+		testStyles(otherTarget.querySelectorAll( 'p' ));
 	}
 };
diff --git a/test/generator/samples/render-in-iframe/main.html b/test/generator/samples/render-in-iframe/main.html
index c67978fa51..5dbaac8827 100644
--- a/test/generator/samples/render-in-iframe/main.html
+++ b/test/generator/samples/render-in-iframe/main.html
@@ -1,6 +1,18 @@
-<h1>Just some static HTML</h1>
+<p>control</p>
+<div>
+	<Widget/>
+</div>
+
 <style>
-h1 {
-	color: rgb(255, 0, 0);
-}
+	p {
+		color: blue;
+	}
 </style>
+
+<script>
+	import Widget from './Widget.html';
+
+	export default {
+		components: { Widget }
+	};
+</script>