From e060f617370440ea78695f52fdc804241cf990ad Mon Sep 17 00:00:00 2001
From: aarshi <aarshi.arora@atlan.com>
Date: Fri, 29 Nov 2024 09:22:04 +0530
Subject: [PATCH] fix return type

---
 .../atlas/repository/store/graph/v2/EntityGraphRetriever.java    | 1 +
 1 file changed, 1 insertion(+)

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 4c7ac69830..ac16006f03 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
@@ -1754,6 +1754,7 @@ public Object getVertexAttributePreFetchCache(AtlasVertex vertex, AtlasAttribute
             Object mappedVertex = mapVertexToAttribute(vertex, attribute, null, false);
             properties.put(attribute.getName(), mappedVertex);
             RequestContext.get().endMetricRecord(nonPrimitiveAttributes);
+            return mappedVertex;
         }
 
         return null;