diff --git a/vite-hardhat/packages/vite/hooks/useOffChainVerification.tsx b/vite-hardhat/packages/vite/hooks/useOffChainVerification.tsx
index 3e249c6..3e0f2a3 100644
--- a/vite-hardhat/packages/vite/hooks/useOffChainVerification.tsx
+++ b/vite-hardhat/packages/vite/hooks/useOffChainVerification.tsx
@@ -14,7 +14,7 @@ export function useOffChainVerification(
   useEffect(() => {
     if (!proofData || !noir) return;
 
-    toast.promise(backend.verifyProof(proofData.proof), {
+    toast.promise(backend.verifyProof(proofData), {
       pending: 'Verifying proof off-chain',
       success: 'Proof verified off-chain',
       error: 'Error verifying proof off-chain',
diff --git a/vite-hardhat/packages/vite/hooks/useProofGeneration.tsx b/vite-hardhat/packages/vite/hooks/useProofGeneration.tsx
index f06b91a..d2d97a9 100644
--- a/vite-hardhat/packages/vite/hooks/useProofGeneration.tsx
+++ b/vite-hardhat/packages/vite/hooks/useProofGeneration.tsx
@@ -3,9 +3,10 @@ import { useEffect, useState } from 'react';
 import { getCircuit } from '../../noir/compile.js';
 import { UltraPlonkBackend } from '@aztec/bb.js';
 import { Noir } from '@noir-lang/noir_js';
+import { ProofData } from '@noir-lang/types';
 
 export function useProofGeneration(inputs?: { [key: string]: string }) {
-  const [proofData, setProofData] = useState<Uint8Array | undefined>();
+  const [proofData, setProofData] = useState<ProofData | undefined>();
   const [backend, setBackend] = useState<UltraPlonkBackend>();
   const [noir, setNoir] = useState<Noir | undefined>();