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;
     }