From 489db812b518cf77367f41b05da21d985be7a427 Mon Sep 17 00:00:00 2001
From: Aseem Bansal <asmbansal2@gmail.com>
Date: Tue, 20 May 2025 10:09:22 +0530
Subject: [PATCH] fix(hook): collect write mutation hook to ensure side effects
 (#13554)

---
 .../java/com/linkedin/metadata/aspect/batch/AspectsBatch.java   | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/entity-registry/src/main/java/com/linkedin/metadata/aspect/batch/AspectsBatch.java b/entity-registry/src/main/java/com/linkedin/metadata/aspect/batch/AspectsBatch.java
index 57f3d481af..3188957925 100644
--- a/entity-registry/src/main/java/com/linkedin/metadata/aspect/batch/AspectsBatch.java
+++ b/entity-registry/src/main/java/com/linkedin/metadata/aspect/batch/AspectsBatch.java
@@ -91,7 +91,7 @@ public interface AspectsBatch {
       Collection<ChangeMCP> changeMCPS, @Nonnull RetrieverContext retrieverContext) {
     for (MutationHook mutationHook :
         retrieverContext.getAspectRetriever().getEntityRegistry().getAllMutationHooks()) {
-      mutationHook.applyWriteMutation(changeMCPS, retrieverContext);
+      mutationHook.applyWriteMutation(changeMCPS, retrieverContext).count();
     }
   }