diff --git a/programs/zk-token-proof/src/lib.rs b/programs/zk-token-proof/src/lib.rs
index d350e3a897..e52f6520d3 100644
--- a/programs/zk-token-proof/src/lib.rs
+++ b/programs/zk-token-proof/src/lib.rs
@@ -134,6 +134,11 @@ declare_process_instruction!(process_instruction, 0, |invoke_context| {
 
     match instruction {
         ProofInstruction::CloseContextState => {
+            if native_programs_consume_cu {
+                invoke_context
+                    .consume_checked(3_300)
+                    .map_err(|_| InstructionError::ComputationalBudgetExceeded)?;
+            }
             ic_msg!(invoke_context, "CloseContextState");
             process_close_proof_context(invoke_context)
         }