From cf12b63a40c5ec05ce95ca21ba50bd6c860ecef7 Mon Sep 17 00:00:00 2001
From: sriram-atlan <sriram.aravamuthan@atlan.com>
Date: Thu, 28 Nov 2024 17:06:02 +0530
Subject: [PATCH] use attribute.getName() in properties Map

---
 .../atlas/repository/store/graph/v2/EntityGraphRetriever.java | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/EntityGraphRetriever.java b/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/EntityGraphRetriever.java
index ff6e19bf8d..d4fb44efee 100644
--- a/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/EntityGraphRetriever.java
+++ b/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/EntityGraphRetriever.java
@@ -1984,8 +1984,8 @@ public Object getVertexAttributePreFetchCache(AtlasVertex vertex, AtlasAttribute
             return null;
         }
 
-        if (properties.get(attribute) != null) {
-            return properties.get(attribute);
+        if (properties.get(attribute.getName()) != null) {
+            return properties.get(attribute.getName());
         }
         return null;
     }