diff --git a/src/helpers/heritage-integration/order-helper.spec.js b/src/helpers/heritage-integration/order-helper.spec.js
index 322d9237d..daf9af6bb 100644
--- a/src/helpers/heritage-integration/order-helper.spec.js
+++ b/src/helpers/heritage-integration/order-helper.spec.js
@@ -179,7 +179,7 @@ describe("saveSession", () => {
         // Assert
         expect(mocks.baseMixin.methods.dispatchStoreActionWithLogging).toHaveBeenCalledWith(
             storeActions.SAVE_SESSION,
-            null,
+            { createDeleteStatusWorkOrderForPia: false, submitAfterSave: false },
             "test"
         );
         expect(mocks.baseMixin.methods.dispatchStoreAction).toHaveBeenCalledWith(
@@ -282,12 +282,12 @@ describe("submitWorkOrder", () => {
         const mocks = setupMocksForJsFiles(mockData);
 
         // Act
-        await submitWorkOrder({ pageNameToLog: "test" });
+        await submitWorkOrder({ pageNameToLog: "test", submitAfterSave: true });
 
         // Assert
         expect(mocks.baseMixin.methods.dispatchStoreActionWithLogging).toHaveBeenCalledWith(
             storeActions.SAVE_SESSION,
-            null,
+            { createDeleteStatusWorkOrderForPia: false, submitAfterSave: true },
             "test"
         );
         expect(mocks.baseMixin.methods.dispatchStoreAction).toHaveBeenCalledWith(
@@ -345,7 +345,7 @@ describe("submitWorkOrder", () => {
         setupCookies({ funnelCookieValue: JSON.stringify(testCookieValue) });
 
         // Act
-        await submitWorkOrder({ pageNameToLog: "test" });
+        await submitWorkOrder({ pageNameToLog: "test", submitAfterSave: true });
 
         // Assert
         expect(cookieHelper.getFunnelCookie().DidHeritageFunnelUpdateLast).toEqual(false);