diff --git a/src/layouts/customer-details/customer-details.spec.js b/src/layouts/customer-details/customer-details.spec.js
index af43c1930..14fb45233 100644
--- a/src/layouts/customer-details/customer-details.spec.js
+++ b/src/layouts/customer-details/customer-details.spec.js
@@ -4,11 +4,6 @@ import customerDetails from "@/layouts/customer-details/customer-details.vue";
 // Supporting Files
 import { shallowMount } from "@vue/test-utils";
 import { getMountOptions } from "@/helpers/unit-test-helper.js";
-import * as navigateToHeritage from "@/helpers/heritage-integration/navigation-helper";
-
-jest.mock("@/helpers/heritage-integration/navigation-helper", () => ({
-    navigateToHeritageFunnel: jest.fn(),
-}));
 
 // Mock our module for promises.
 jest.mock("@/helpers/layout-helper.js", () => ({
@@ -36,7 +31,7 @@ describe("customer-details.vue", () => {
             await wrapper.vm.forwardButtonAction();
 
             // Assert
-            expect(navigateToHeritage.navigateToHeritageFunnel).toHaveBeenCalled();
+            expect(wrapper.vm.$router.navigateWithSaving).toHaveBeenCalled();
         });
     });
 });