diff --git a/src/tools/flash_close_trade.ts b/src/tools/flash_close_trade.ts
index 3a1e4e9..c30d0b6 100644
--- a/src/tools/flash_close_trade.ts
+++ b/src/tools/flash_close_trade.ts
@@ -94,7 +94,7 @@ export async function flashCloseTrade(
       priceWithSlippage,
       sideEnum,
       poolConfig,
-      get_flash_privilege(),
+      get_flash_privilege(agent),
       tradingAccounts.nftTradingAccountPk,
       tradingAccounts.nftReferralAccountPK,
       tradingAccounts.nftOwnerRebateTokenAccountPk,
diff --git a/src/tools/flash_open_trade.ts b/src/tools/flash_open_trade.ts
index 8e1af4e..6a646dc 100644
--- a/src/tools/flash_open_trade.ts
+++ b/src/tools/flash_open_trade.ts
@@ -141,7 +141,7 @@ export async function flashOpenTrade(
       positionSize,
       side === "long" ? Side.Long : Side.Short,
       poolConfig,
-      get_flash_privilege(),
+      get_flash_privilege(agent),
       tradingAccounts.nftTradingAccountPk,
       tradingAccounts.nftReferralAccountPK,
       tradingAccounts.nftOwnerRebateTokenAccountPk!,
diff --git a/src/utils/flashUtils.ts b/src/utils/flashUtils.ts
index a583690..5008d85 100644
--- a/src/utils/flashUtils.ts
+++ b/src/utils/flashUtils.ts
@@ -10,6 +10,7 @@ import {
 } from "flash-sdk";
 import { Cluster, PublicKey, Connection, Keypair } from "@solana/web3.js";
 import { getAssociatedTokenAddressSync } from "@solana/spl-token";
+import { SolanaAgentKit } from "../index";
 
 const POOL_NAMES = [
   "Crypto.1",
@@ -285,9 +286,9 @@ export function createPerpClient(
   );
 }
 
-const FLASH_PRIVILEGE = process.env.FLASH_PRIVILEGE || "None";
+export function get_flash_privilege(agent: SolanaAgentKit): Privilege {
+  const FLASH_PRIVILEGE = agent.config.FLASH_PRIVILEGE || "None";
 
-export function get_flash_privilege(): Privilege {
   switch (FLASH_PRIVILEGE.toLowerCase()) {
     case "referral":
       return Privilege.Referral;