diff --git a/packages/js/client/src/__tests__/e2e/workflow.spec.ts b/packages/js/client/src/__tests__/e2e/workflow.spec.ts
index b6f8745424..a8f557b98d 100644
--- a/packages/js/client/src/__tests__/e2e/workflow.spec.ts
+++ b/packages/js/client/src/__tests__/e2e/workflow.spec.ts
@@ -6,7 +6,7 @@ import {
   ensAddresses,
   providers
 } from "@polywrap/test-env-js";
-import { createPolywrapClient, PolywrapClient, PolywrapClientConfig } from "../..";
+import { createPolywrapClient, JobResult, PolywrapClient, PolywrapClientConfig } from "../..";
 import { outPropWorkflow, sanityWorkflow } from "./workflow-test-cases";
 
 jest.setTimeout(200000);
@@ -89,8 +89,8 @@ describe("workflow", () => {
     test("sanity workflow", async () => {
       await client.run({
         workflow: sanityWorkflow,
-        onExecution: async (id: string, data: unknown, error: unknown) => {
-          await tests[id](data, error);
+        onExecution: async (id: string, jobResult: JobResult) => {
+          await tests[id](jobResult.data, jobResult.error);
         },
       });
     });
@@ -98,8 +98,8 @@ describe("workflow", () => {
     test("workflow with output propagation", async () => {
       await client.run({
         workflow: outPropWorkflow,
-        onExecution: async (id: string, data: unknown, error: unknown) => {
-          await tests[id](data, error);
+        onExecution: async (id: string, jobResult: JobResult) => {
+          await tests[id](jobResult.data, jobResult.error);
         },
       });
     });