diff --git a/src/main/java/com/flowingcode/vaadin/addons/demo/TabbedDemo.java b/src/main/java/com/flowingcode/vaadin/addons/demo/TabbedDemo.java
index df5cc59..635e015 100644
--- a/src/main/java/com/flowingcode/vaadin/addons/demo/TabbedDemo.java
+++ b/src/main/java/com/flowingcode/vaadin/addons/demo/TabbedDemo.java
@@ -361,9 +361,9 @@ public static void applyTheme(Element element, String theme) {
 
     String script;
     if (element.getTag().equalsIgnoreCase("iframe")) {
-      script = "this.contentWindow.document.body.setAttribute('theme', $0)";
+      script = "this.contentWindow.document.documentElement.setAttribute('theme', $0)";
     } else {
-      script = "document.body.setAttribute('theme', $0);";
+      script = "document.documentElement.setAttribute('theme', $0);";
     }
 
     element.executeJs(script, theme);