diff --git a/card-tools.js b/card-tools.js
index 5f00ac6..1db3785 100644
--- a/card-tools.js
+++ b/card-tools.js
@@ -12,7 +12,7 @@ class {
   }
 
   static litElement() {
-    return Object.getPrototypeOf(customElements.get('hui-error-entity-row'));
+    return Object.getPrototypeOf(customElements.get('home-assistant-main'));
   }
 
   static litHtml() {