From bff8a650bc1d7d7c82113b44e4f85908ac7898b0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Thomas=20Lov=C3=A9n?= <thomasloven@gmail.com>
Date: Fri, 26 Jul 2019 17:51:13 +0200
Subject: [PATCH] Add fix to work better (i.e. not necessarily fully) with Home
 Assistant Cast

---
 card-tools.js | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/card-tools.js b/card-tools.js
index fe7be08..364522c 100644
--- a/card-tools.js
+++ b/card-tools.js
@@ -19,7 +19,9 @@ class {
   }
 
   static get LitElement() {
-    return Object.getPrototypeOf(customElements.get('home-assistant-main'));
+    if(customElements.get('home-assistant-main'))
+      return Object.getPrototypeOf(customElements.get('home-assistant-main'));
+    return Object.getPrototypeOf(customElements.get('hui-view'));
   }
   static litElement() { // Backwards compatibility - deprecated
     this.deprecationWarning();