use attribute.getName() in properties Map

atlanhq · Nov 28, 2024 · cf12b63 · cf12b63
1 parent 937c62e
commit cf12b63
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...sitory/src/main/java/org/apache/atlas/repository/store/graph/v2/EntityGraphRetriever.java b/...sitory/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;
     }