From 37cb5fdb7547463522d65835f0b992553ab5e4b8 Mon Sep 17 00:00:00 2001
From: Jan-Philipp Benecke <jan-philipp@bnck.me>
Date: Thu, 12 Dec 2024 19:47:52 +0100
Subject: [PATCH] Fix rebase

---
 src/panels/config/automation/ha-automation-editor.ts | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/src/panels/config/automation/ha-automation-editor.ts b/src/panels/config/automation/ha-automation-editor.ts
index f7e6c068066a..e5a72c77008a 100644
--- a/src/panels/config/automation/ha-automation-editor.ts
+++ b/src/panels/config/automation/ha-automation-editor.ts
@@ -22,7 +22,6 @@ import type { CSSResultGroup, PropertyValues, TemplateResult } from "lit";
 import { LitElement, css, html, nothing } from "lit";
 import { property, state } from "lit/decorators";
 import { classMap } from "lit/directives/class-map";
-import memoizeOne from "memoize-one";
 import { consume } from "@lit-labs/context";
 import { fireEvent } from "../../../common/dom/fire_event";
 import { navigate } from "../../../common/navigate";
@@ -54,9 +53,8 @@ import { substituteBlueprint } from "../../../data/blueprint";
 import { validateConfig } from "../../../data/config";
 import { UNAVAILABLE } from "../../../data/entity";
 import {
-  fetchEntityRegistry,
   type EntityRegistryEntry,
-    updateEntityRegistryEntry,
+  updateEntityRegistryEntry,
 } from "../../../data/entity_registry";
 import {
   showAlertDialog,
@@ -888,9 +886,7 @@ export class HaAutomationEditor extends PreventUnsavedMixin(
         },
         onClose: () => resolve(false),
         entityRegistryUpdate: this._entityRegistryUpdate,
-        entityRegistryEntry: this.entityRegistry.find(
-          (entry) => entry.unique_id === this.automationId
-        ),
+        entityRegistryEntry: this._registryEntry,
       });
     });
   }