From eea4b65b801e2c0c06e9b17e81e16a519953d9ab Mon Sep 17 00:00:00 2001
From: Mayur Singal <39544459+ulixius9@users.noreply.github.com>
Date: Mon, 27 Jan 2025 15:54:41 +0530
Subject: [PATCH] Fix #19198: di fix custom property filter (#19531)

---
 .../processors/DataInsightsEntityEnricherProcessor.java         | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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 c54ada72491..840b7694fc5 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
@@ -227,7 +227,7 @@ public class DataInsightsEntityEnricherProcessor
     }
 
     // Modify Custom Property key
-    Optional<Object> oCustomProperties = Optional.ofNullable(entityMap.remove("extension"));
+    Optional<Object> oCustomProperties = Optional.ofNullable(entityMap.get("extension"));
     oCustomProperties.ifPresent(
         o -> entityMap.put(String.format("%sCustomProperty", entityType), o));