diff --git a/nifi/nifi-mock/src/main/java/org/apache/nifi/util/MockProcessSession.java b/nifi/nifi-mock/src/main/java/org/apache/nifi/util/MockProcessSession.java
index 13198dc0bb..a5f9f3c7bf 100644
--- a/nifi/nifi-mock/src/main/java/org/apache/nifi/util/MockProcessSession.java
+++ b/nifi/nifi-mock/src/main/java/org/apache/nifi/util/MockProcessSession.java
@@ -66,7 +66,7 @@ public class MockProcessSession implements ProcessSession {
     private final Map<Long, MockFlowFile> originalVersions = new HashMap<>();
     private final SharedSessionState sharedState;
     private final Map<String, Long> counterMap = new HashMap<>();
-    private final ProvenanceReporter provenanceReporter;
+    private final MockProvenanceReporter provenanceReporter;
 
     private boolean committed = false;
     private boolean rolledback = false;
@@ -135,6 +135,7 @@ public class MockProcessSession implements ProcessSession {
             sharedState.adjustCounter(entry.getKey(), entry.getValue());
         }
 
+        sharedState.addProvenanceEvents(provenanceReporter.getEvents());
         counterMap.clear();
     }