diff --git a/openmetadata-service/src/main/java/org/openmetadata/service/apps/bundles/insights/workflows/dataAssets/processors/DataInsightsEntityEnricherProcessor.java b/openmetadata-service/src/main/java/org/openmetadata/service/apps/bundles/insights/workflows/dataAssets/processors/DataInsightsEntityEnricherProcessor.java
index 834528f58d2..abf888124c4 100644
--- a/openmetadata-service/src/main/java/org/openmetadata/service/apps/bundles/insights/workflows/dataAssets/processors/DataInsightsEntityEnricherProcessor.java
+++ b/openmetadata-service/src/main/java/org/openmetadata/service/apps/bundles/insights/workflows/dataAssets/processors/DataInsightsEntityEnricherProcessor.java
@@ -7,7 +7,6 @@ import static org.openmetadata.service.workflows.searchIndex.ReindexingUtil.ENTI
 import static org.openmetadata.service.workflows.searchIndex.ReindexingUtil.TIMESTAMP_KEY;
 import static org.openmetadata.service.workflows.searchIndex.ReindexingUtil.getUpdatedStats;
 
-import java.util.Collection;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
@@ -52,11 +51,14 @@ public class DataInsightsEntityEnricherProcessor
     try {
       enrichedMaps =
           input.getData().stream()
-              .map(entity -> getEntityVersions(entity, contextData))
-              .flatMap(Collection::stream)
-              .map(entityVersionMap -> enrichEntity(entityVersionMap, contextData))
-              .map(this::generateDailyEntitySnapshots)
-              .flatMap(Collection::stream)
+              .flatMap(
+                  entity ->
+                      getEntityVersions(entity, contextData).stream()
+                          .flatMap(
+                              entityVersionMap ->
+                                  generateDailyEntitySnapshots(
+                                      enrichEntity(entityVersionMap, contextData))
+                                      .stream()))
               .toList();
       updateStats(input.getData().size(), 0);
     } catch (Exception e) {