From cd961c25d181fb907c1481e9af6dce6fed375552 Mon Sep 17 00:00:00 2001
From: TA31OU <kyle.opperman@ing.com>
Date: Wed, 3 Sep 2025 12:17:06 +0200
Subject: [PATCH] Revert useEntityStore change

Signed-off-by: TA31OU <kyle.opperman@ing.com>
---
 .changeset/flat-peas-run.md                                     | 2 +-
 .../src/components/EntityRelationsGraph/useEntityStore.ts       | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/.changeset/flat-peas-run.md b/.changeset/flat-peas-run.md
index 116e8eb449..c09eadecf8 100644
--- a/.changeset/flat-peas-run.md
+++ b/.changeset/flat-peas-run.md
@@ -3,4 +3,4 @@
 '@backstage/plugin-catalog-graph': minor
 ---
 
-Added `renderEdge` prop to <DependencyGraph /> component in @backstage/core-components to allow custom rendering of graph edges. Fixed local development for EntityRelationsGraph plugin.
+Added `renderEdge` prop to <DependencyGraph /> component in @backstage/core-components to allow custom rendering of graph edges.
diff --git a/plugins/catalog-graph/src/components/EntityRelationsGraph/useEntityStore.ts b/plugins/catalog-graph/src/components/EntityRelationsGraph/useEntityStore.ts
index be6ea81f57..c55becdd49 100644
--- a/plugins/catalog-graph/src/components/EntityRelationsGraph/useEntityStore.ts
+++ b/plugins/catalog-graph/src/components/EntityRelationsGraph/useEntityStore.ts
@@ -64,7 +64,7 @@ export function useEntityStore(): {
       return;
     }
 
-    const { items } = await catalogClient.getEntities();
+    const { items } = await catalogClient.getEntitiesByRefs({ entityRefs });
     items.forEach(ent => {
       if (ent) {
         const entityRef = stringifyEntityRef(ent);