diff --git a/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/preprocessor/contract/ContractPreProcessor.java b/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/preprocessor/contract/ContractPreProcessor.java
index bc9ff76cbd..98add96a39 100644
--- a/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/preprocessor/contract/ContractPreProcessor.java
+++ b/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/preprocessor/contract/ContractPreProcessor.java
@@ -192,6 +192,7 @@ private void updateExistingVersion(EntityMutationContext context, AtlasEntity en
         AtlasEntityType entityType = ensureEntityType(entity.getTypeName());
 
         context.addUpdated(entity.getGuid(), entity, entityType, vertex);
+        recordEntityMutatedDetails(context, entity, vertex);
 
     }
 
@@ -283,11 +284,14 @@ private void datasetAttributeSync(EntityMutationContext context, AtlasEntity ass
 
         AtlasVertex vertex = AtlasGraphUtilsV2.findByGuid(entity.getGuid());
         AtlasEntityType entityType = ensureEntityType(entity.getTypeName());
+        context.addUpdated(entity.getGuid(), entity, entityType, vertex);
+        recordEntityMutatedDetails(context, entity, vertex);
+    }
+
+    private void recordEntityMutatedDetails(EntityMutationContext context, AtlasEntity entity, AtlasVertex vertex) throws AtlasBaseException {
         AtlasEntityComparator entityComparator = new AtlasEntityComparator(typeRegistry, entityRetriever, context.getGuidAssignments(), true, true);
         AtlasEntityComparator.AtlasEntityDiffResult diffResult   = entityComparator.getDiffResult(entity, vertex, !storeDifferentialAudits);
         RequestContext        reqContext           = RequestContext.get();
-        context.addUpdated(entity.getGuid(), entity, entityType, vertex);
-
         if (diffResult.hasDifference()) {
             if (storeDifferentialAudits) {
                 diffResult.getDiffEntity().setGuid(entity.getGuid());