diff --git a/package.json b/package.json
index 0b3c903..b43b68c 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "flower-card",
-  "version": "2024.9.0-beta1",
+  "version": "2024.9.0-beta2",
   "description": "Custom flower card for https://github.com/Olen/homeassistant-plant",
   "keywords": [
     "home-assistant",
diff --git a/src/editor.ts b/src/editor.ts
index 1014fed..8e9d71b 100644
--- a/src/editor.ts
+++ b/src/editor.ts
@@ -19,8 +19,8 @@ export class FlowerCardEditor extends EditorForm {
             this._config.show_bars = default_show_bars;
         }
 
-        const plantsList = getEntitiesByDomain('plant');
-        const batteryList = getEntitiesByDeviceClass("sensor", "battery");
+        const plantsList = getEntitiesByDomain(this._hass, 'plant');
+        const batteryList = getEntitiesByDeviceClass(this._hass, "sensor", "battery");
 
         return this.renderForm([
             { controls: [{ label: "Display Type", configValue: "display_type", type: FormControlType.Radio, items: [