Update compact-custom-header.js

maykar · Mar 16, 2019 · 094682d · 094682d
1 parent 17069fd
commit 094682d
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/compact-custom-header.js b/compact-custom-header.js
@@ -284,12 +284,21 @@ if (!customElements.get("compact-custom-header")) {
 
     get rootElement() {
       try {
-        return document
+        let panelResolver = document
           .querySelector("home-assistant")
           .shadowRoot.querySelector("home-assistant-main")
           .shadowRoot.querySelector("app-drawer-layout partial-panel-resolver")
-          .shadowRoot.querySelector("ha-panel-lovelace")
-          .shadowRoot.querySelector("hui-root").shadowRoot;
+        if (panelResolver.shadowRoot){
+          return panelResolver
+            .shadowRoot.querySelector("ha-panel-lovelace")
+            .shadowRoot.querySelector("hui-root").shadowRoot;
+        } else {
+          return document
+            .querySelector("home-assistant")
+            .shadowRoot.querySelector("home-assistant-main")
+            .shadowRoot.querySelector("ha-panel-lovelace")
+            .shadowRoot.querySelector("hui-root").shadowRoot;
+        }
       } catch(e) {
         console.log("Can't find 'hui-root', going to walk the DOM to find it.");
       }