From a8cdcb54f3f93be2618faeaa51225a1a20df6838 Mon Sep 17 00:00:00 2001
From: Sewmina Dilshan <sewmina7@gmail.com>
Date: Fri, 20 Jun 2025 17:48:58 +0530
Subject: [PATCH] close bet token, set correct balance

---
 programs/bets/src/instructions/close_bet_token.rs | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/programs/bets/src/instructions/close_bet_token.rs b/programs/bets/src/instructions/close_bet_token.rs
index 34c0db4..e93cf65 100644
--- a/programs/bets/src/instructions/close_bet_token.rs
+++ b/programs/bets/src/instructions/close_bet_token.rs
@@ -17,11 +17,15 @@ pub fn close_token_bet(ctx: Context<CloseBetToken>, winner: Pubkey, userid: Stri
         bet_vault.game_id.as_bytes(),
         &bet_vault.nonce.to_le_bytes(),
     ];
-    // Transfer tokens from vault to winner
+    
+    // Get the current token balance of the bet vault
+    let token_balance = ctx.accounts.token_vault.amount;
+    
+    // Transfer all tokens from vault to winner
     transfer_tokens_pda(
         &ctx.accounts.token_vault,
         &ctx.accounts.winner_token_account,
-        bet_vault.wager,
+        token_balance,
         &ctx.accounts.token_mint,
         &bet_vault,
         &ctx.accounts.token_program,
@@ -35,7 +39,7 @@ pub fn close_token_bet(ctx: Context<CloseBetToken>, winner: Pubkey, userid: Stri
         bets_list.bets.remove(pos);
     }
 
-    msg!("Bet closed and {} tokens transferred to winner {}", bet_vault.wager, winner);
+    msg!("Bet closed and {} tokens transferred to winner {}", token_balance, winner);
     Ok(())
 }