use strict treasury_tokens loader

2026-05-18 23:16:48 +00:00 · 2024-07-02 13:08:35 +00:00
parent b55b6d338d
commit 3b1039b469
5 changed files with 31 additions and 20 deletions
--- a/program/src/stake.rs
+++ b/program/src/stake.rs
@@ -30,12 +30,7 @@ pub fn process_stake<'a, 'info>(
    load_signer(signer)?;
    load_proof(proof_info, signer.key, true)?;
    load_token_account(sender_info, Some(signer.key), &MINT_ADDRESS, true)?;
-    load_token_account(
-        treasury_tokens_info,
-        Some(&TREASURY_ADDRESS),
-        &MINT_ADDRESS,
-        true,
-    )?;
+    load_treasury_tokens(treasury_tokens_info, true)?;
    load_program(token_program, spl_token::id())?;

    // Update proof balance