diff --git a/dist/websnark.d.ts b/dist/websnark.d.ts
index 95b953d..9a415bd 100644
--- a/dist/websnark.d.ts
+++ b/dist/websnark.d.ts
@@ -11,9 +11,17 @@ export type snarkInputs = {
     pathElements: Element[];
     pathIndices: Element[];
 };
+export type snarkArgs = [
+    _root: string,
+    _nullifierHash: string,
+    _recipient: string,
+    _relayer: string,
+    _fee: string,
+    _refund: string
+];
 export type snarkProofs = {
     proof: string;
-    args: string[];
+    args: snarkArgs;
 };
 export declare function initGroth16(): Promise<void>;
 export declare function calculateSnarkProof(input: snarkInputs, circuit: object, provingKey: ArrayBuffer): Promise<snarkProofs>;
diff --git a/src/websnark.ts b/src/websnark.ts
index 72a88ab..ecc1072 100644
--- a/src/websnark.ts
+++ b/src/websnark.ts
@@ -21,9 +21,18 @@ export type snarkInputs = {
   pathIndices: Element[];
 };
 
+export type snarkArgs = [
+  _root: string,
+  _nullifierHash: string,
+  _recipient: string,
+  _relayer: string,
+  _fee: string,
+  _refund: string,
+];
+
 export type snarkProofs = {
   proof: string;
-  args: string[];
+  args: snarkArgs;
 };
 
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
@@ -71,7 +80,7 @@ export async function calculateSnarkProof(
     input.relayer,
     toFixedHex(input.fee, 32),
     toFixedHex(input.refund, 32),
-  ];
+  ] as snarkArgs;
 
   return { proof, args };
 }