diff --git a/button-entity-row.js b/button-entity-row.js
index b1f6c60..6486f64 100644
--- a/button-entity-row.js
+++ b/button-entity-row.js
@@ -1,4 +1,6 @@
-import { LitElement, html, css } from "https://unpkg.com/lit-element@2.0.1/lit-element.js?module"
+const LitElement = Object.getPrototypeOf(customElements.get("hui-view"))
+const html = LitElement.prototype.html
+const css = LitElement.prototype.css
 
 class ButtonEntityRow extends LitElement {
   static get properties() {
@@ -86,16 +88,16 @@ class ButtonEntityRow extends LitElement {
         let button =
           typeof item === "string"
             ? {
-              entityId: item,
-              icon: undefined,
-              stateIcons: undefined,
-              stateStyles: undefined,
-              stateIconStyles: undefined,
-              style: undefined,
-              iconStyle: undefined,
-              name: undefined,
-              service: undefined,
-              serviceData: undefined
+                entityId: item,
+                icon: undefined,
+                stateIcons: undefined,
+                stateStyles: undefined,
+                stateIconStyles: undefined,
+                style: undefined,
+                iconStyle: undefined,
+                name: undefined,
+                service: undefined,
+                serviceData: undefined
               }
             : {
                 entityId: item.entity,