diff --git a/button-entity-row.js b/button-entity-row.js
index 8b16d48..1222dc6 100644
--- a/button-entity-row.js
+++ b/button-entity-row.js
@@ -1,6 +1,8 @@
-const LitElement = Object.getPrototypeOf(customElements.get("hui-view"))
-const html = LitElement.prototype.html
-const css = LitElement.prototype.css
+import {
+  LitElement,
+  html,
+  css,
+} from "https://unpkg.com/lit-element@2.0.1/lit-element.js?module";
 
 class ButtonEntityRow extends LitElement {
   static get properties() {