diff --git a/dist/google-keep-card.js b/dist/google-keep-card.js
index 2b5e83e..31fd66c 100644
--- a/dist/google-keep-card.js
+++ b/dist/google-keep-card.js
@@ -1,7 +1,5 @@
-const LitElement = Object.getPrototypeOf(
-    customElements.get("ha-panel-lovelace")
-);
-const html = LitElement.prototype.html;
+var LitElement = LitElement || Object.getPrototypeOf(customElements.get("home-assistant-main"));
+var html = LitElement.prototype.html;
 
 const themeLight = "light";
 const themeDark = "dark";