diff --git a/src/lib/proof-system/zkprogram.ts b/src/lib/proof-system/zkprogram.ts
index 5495e3a6ab..257d0edb35 100644
--- a/src/lib/proof-system/zkprogram.ts
+++ b/src/lib/proof-system/zkprogram.ts
@@ -909,6 +909,7 @@ function picklesRuleFromFunction(
         result = (await func(...finalArgs)) as any;
       } else {
         let input = fromFieldVars(publicInputType, publicInput, auxInputData);
+        publicInputType.check(input);
         result = (await func(input, ...finalArgs)) as any;
       }
       proofs = ZkProgramContext.getDeclaredProofs();