From 094682da85f5ce91c08bfa2e791acf59bdcdfee5 Mon Sep 17 00:00:00 2001
From: Ryan Meek <contact@ninjacyb.org>
Date: Fri, 15 Mar 2019 23:06:54 -0400
Subject: [PATCH] Update compact-custom-header.js

---
 compact-custom-header.js | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/compact-custom-header.js b/compact-custom-header.js
index ba66f1b..7006085 100644
--- 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.");
       }