diff --git a/src/lib/provable/test/arithmetic.unit-test.ts b/src/lib/provable/test/arithmetic.unit-test.ts
index 248529b0e..ac40d0b10 100644
--- a/src/lib/provable/test/arithmetic.unit-test.ts
+++ b/src/lib/provable/test/arithmetic.unit-test.ts
@@ -27,7 +27,7 @@ let Arithmetic = ZkProgram({
     divMod64: {
       privateInputs: [Field],
       async method(a: Field) {
-        return Gadgets.divMod64(a);
+        return { publicOutput: Gadgets.divMod64(a) };
       },
     },
   },
@@ -93,6 +93,6 @@ await equivalentAsync({ from: [field], to: divModOutput }, { runs: 3 })(
     return divMod64Helper(x);
   },
   async (x) => {
-    return (await Arithmetic.divMod64(x)).publicOutput;
+    return (await Arithmetic.divMod64(x)).proof.publicOutput;
   }
 );
diff --git a/src/lib/provable/test/bitwise.unit-test.ts b/src/lib/provable/test/bitwise.unit-test.ts
index 33b7fe18f..2f4efb3b4 100644
--- a/src/lib/provable/test/bitwise.unit-test.ts
+++ b/src/lib/provable/test/bitwise.unit-test.ts
@@ -60,7 +60,7 @@ let Bitwise = ZkProgram({
     or: {
       privateInputs: [Field, Field],
       async method(a: Field, b: Field) {
-        return Gadgets.or(a, b, 64);
+        return { publicOutput: Gadgets.or(a, b, 64) };
       },
     },
     rot32: {
diff --git a/src/lib/provable/test/blake2b.unit-test.ts b/src/lib/provable/test/blake2b.unit-test.ts
index d18019823..a15589980 100644
--- a/src/lib/provable/test/blake2b.unit-test.ts
+++ b/src/lib/provable/test/blake2b.unit-test.ts
@@ -38,7 +38,7 @@ const BLAKE2BProgram = ZkProgram({
     blake2b: {
       privateInputs: [Bytes(192)],
       async method(preImage: Bytes) {
-        return Gadgets.BLAKE2B.hash(preImage);
+        return { publicOutput: Gadgets.BLAKE2B.hash(preImage) };
       },
     },
   },
@@ -55,7 +55,7 @@ await equivalentAsync(
   },
   { runs: RUNS }
 )(nobleBlake2b, async (x) => {
-  const proof = await BLAKE2BProgram.blake2b(x);
+  const { proof } = await BLAKE2BProgram.blake2b(x);
   await BLAKE2BProgram.verify(proof);
   return proof.publicOutput;
 });