diff --git a/programs/zk-token-proof/src/lib.rs b/programs/zk-token-proof/src/lib.rs
index 73417e5dbc..96df4cdc6a 100644
--- a/programs/zk-token-proof/src/lib.rs
+++ b/programs/zk-token-proof/src/lib.rs
@@ -44,13 +44,13 @@ pub fn process_instruction(
             ic_msg!(invoke_context, "VerifyWithdraw");
             verify::<WithdrawData>(input, invoke_context)
         }
-        ProofInstruction::VerifyTransferWithRangeProofData => {
-            ic_msg!(invoke_context, "VerifyTransferWithRangeProofData");
-            verify::<TransferWithRangeProofData>(input, invoke_context)
+        ProofInstruction::VerifyTransferRangeProofData => {
+            ic_msg!(invoke_context, "VerifyTransferRangeProofData");
+            verify::<TransferRangeProofData>(input, invoke_context)
         }
-        ProofInstruction::VerifyTransferWithValidityProofData => {
-            ic_msg!(invoke_context, "VerifyTransferWithValidityProofData");
-            verify::<TransferWithValidityProofData>(input, invoke_context)
+        ProofInstruction::VerifyTransferValidityProofData => {
+            ic_msg!(invoke_context, "VerifyTransferValidityProofData");
+            verify::<TransferValidityProofData>(input, invoke_context)
         }
     }
 }