flash-sdk 2.25.2 → 2.25.4

package/dist/PerpetualsClient.js

@@ -501,7 +501,7 @@ var PerpetualsClient = (function () {
                 }
             });
         }); };
-        this.addCustody = function (poolName, tokenMint, isStable, isVirtual, oracle, pricing, permissions, fees, borrowRate, ratios, depegAdjustment) { return __awaiter(_this, void 0, void 0, function () {
+        this.addCustody = function (poolName, tokenMint, isStable, isVirtual, token22, oracle, pricing, permissions, fees, borrowRate, ratios, depegAdjustment) { return __awaiter(_this, void 0, void 0, function () {
             var trx_id, error_1;
             return __generator(this, function (_a) {
                 switch (_a.label) {
@@ -511,6 +511,7 @@ var PerpetualsClient = (function () {
                                 .addCustody({
                                 isStable: isStable,
                                 isVirtual: isVirtual,
+                                token22: token22,
                                 oracle: oracle,
                                 pricing: pricing,
                                 permissions: permissions,
@@ -2481,34 +2482,33 @@ var PerpetualsClient = (function () {
             if (skipBalanceChecks === void 0) { skipBalanceChecks = false; }
             if (ephemeralSignerPubkey === void 0) { ephemeralSignerPubkey = undefined; }
             return __awaiter(_this, void 0, void 0, function () {
-                var publicKey, targetCustodyConfig, collateralCustodyConfig, marketAccount, userCollateralTokenAccount, wrappedSolAccount, preInstructions, instructions, postInstructions, additionalSigners, lamports, unWrappedSolBalance, _a, tokenAccountBalance, _b, positionAccount, params, instruction;
+                var publicKey, targetCustodyConfig, collateralCustodyConfig, collateralToken, marketAccount, userCollateralTokenAccount, wrappedSolAccount, preInstructions, instructions, postInstructions, additionalSigners, lamports, unWrappedSolBalance, _a, tokenAccountBalance, _b, positionAccount, rebateMintAccount, params, instruction;
                 return __generator(this, function (_c) {
                     switch (_c.label) {
                         case 0:
                             publicKey = this.provider.wallet.publicKey;
                             targetCustodyConfig = poolConfig.custodies.find(function (i) { return i.mintKey.equals(poolConfig.getTokenFromSymbol(targetSymbol).mintKey); });
                             collateralCustodyConfig = poolConfig.custodies.find(function (i) { return i.mintKey.equals(poolConfig.getTokenFromSymbol(collateralSymbol).mintKey); });
+                            collateralToken = poolConfig.getTokenFromSymbol(collateralSymbol);
                             marketAccount = poolConfig.getMarketPk(targetCustodyConfig.custodyAccount, collateralCustodyConfig.custodyAccount, side);
-                            return [4, (0, spl_token_1.getAssociatedTokenAddress)(poolConfig.getTokenFromSymbol(collateralSymbol).mintKey, publicKey, true)];
-                        case 1:
-                            userCollateralTokenAccount = _c.sent();
+                            userCollateralTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.getTokenFromSymbol(collateralSymbol).mintKey, publicKey, true, collateralToken.isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             preInstructions = [];
                             instructions = [];
                             postInstructions = [];
                             additionalSigners = [];
-                            if (!(collateralSymbol == 'SOL')) return [3, 4];
+                            if (!(collateralSymbol == 'SOL')) return [3, 3];
                             console.log("collateralSymbol === SOL", collateralSymbol);
                             lamports = collateralWithfee.add(new anchor_1.BN(this.minimumBalanceForRentExemptAccountLamports));
-                            if (!!skipBalanceChecks) return [3, 3];
+                            if (!!skipBalanceChecks) return [3, 2];
                             _a = anchor_1.BN.bind;
                             return [4, this.provider.connection.getBalance(publicKey)];
-                        case 2:
+                        case 1:
                             unWrappedSolBalance = new (_a.apply(anchor_1.BN, [void 0, _c.sent()]))();
                             if (unWrappedSolBalance.lt(lamports)) {
                                 throw "Insufficient SOL Funds";
                             }
-                            _c.label = 3;
-                        case 3:
+                            _c.label = 2;
+                        case 2:
                             if (!ephemeralSignerPubkey) {
                                 wrappedSolAccount = new web3_js_1.Keypair();
                                 additionalSigners.push(wrappedSolAccount);
@@ -2526,24 +2526,29 @@ var PerpetualsClient = (function () {
                             postInstructions = [
                                 (0, spl_token_1.createCloseAccountInstruction)((ephemeralSignerPubkey ? ephemeralSignerPubkey : wrappedSolAccount.publicKey), publicKey, publicKey),
                             ];
-                            return [3, 7];
-                        case 4:
-                            if (!!skipBalanceChecks) return [3, 7];
+                            return [3, 6];
+                        case 3:
+                            if (!!skipBalanceChecks) return [3, 6];
                             return [4, (0, utils_1.checkIfAccountExists)(userCollateralTokenAccount, this.provider.connection)];
-                        case 5:
+                        case 4:
                             if (!(_c.sent())) {
                                 throw "Insufficient Funds , token Account doesn't exist";
                             }
                             _b = anchor_1.BN.bind;
                             return [4, this.provider.connection.getTokenAccountBalance(userCollateralTokenAccount)];
-                        case 6:
+                        case 5:
                             tokenAccountBalance = new (_b.apply(anchor_1.BN, [void 0, (_c.sent()).value.amount]))();
                             if (tokenAccountBalance.lt(collateralWithfee)) {
                                 throw "Insufficient Funds need more ".concat(collateralWithfee.sub(tokenAccountBalance), " tokens");
                             }
-                            _c.label = 7;
-                        case 7:
+                            _c.label = 6;
+                        case 6:
                             positionAccount = poolConfig.getPositionFromMarketPk(publicKey, marketAccount);
+                            rebateMintAccount = {
+                                pubkey: collateralCustodyConfig.mintKey,
+                                isSigner: false,
+                                isWritable: false
+                            };
                             params = {
                                 priceWithSlippage: priceWithSlippage,
                                 collateralAmount: collateralWithfee,
@@ -2570,11 +2575,12 @@ var PerpetualsClient = (function () {
                                     eventAuthority: this.eventAuthority.publicKey,
                                     program: this.programId,
                                     transferAuthority: this.authority.publicKey,
-                                    ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY
+                                    ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY,
+                                    fundingMint: collateralCustodyConfig.mintKey
                                 })
-                                    .remainingAccounts(__spreadArray([], (0, getNftAccounts_1.getNftAccounts)(nftTradingAccount, nftReferralAccount, nftRebateTokenAccount, privilege), true))
+                                    .remainingAccounts(__spreadArray(__spreadArray([], (0, getNftAccounts_1.getNftAccounts)(nftTradingAccount, nftReferralAccount, nftRebateTokenAccount, privilege), true), [rebateMintAccount], false))
                                     .instruction()];
-                        case 8:
+                        case 7:
                             instruction = _c.sent();
                             instructions.push(instruction);
                             return [2, {
@@ -2640,7 +2646,7 @@ var PerpetualsClient = (function () {
                             ];
                             return [3, 6];
                         case 3:
-                            userInputTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(swapPoolConfig.getTokenFromSymbol(inputSymbol).mintKey, publicKey, true);
+                            userInputTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(swapPoolConfig.getTokenFromSymbol(inputSymbol).mintKey, publicKey, true, swapPoolConfig.getTokenFromSymbol(inputSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             if (!!skipBalanceChecks) return [3, 6];
                             return [4, (0, utils_1.checkIfAccountExists)(userInputTokenAccount, this.provider.connection)];
                         case 4:
@@ -2656,11 +2662,11 @@ var PerpetualsClient = (function () {
                             }
                             _d.label = 6;
                         case 6:
-                            userOutputTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(swapPoolConfig.getTokenFromSymbol(swapOutCustodyConfig.symbol).mintKey, publicKey, true);
+                            userOutputTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(swapPoolConfig.getTokenFromSymbol(swapOutCustodyConfig.symbol).mintKey, publicKey, true, swapPoolConfig.getTokenFromSymbol(swapOutCustodyConfig.symbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             return [4, (0, utils_1.checkIfAccountExists)(userOutputTokenAccount, this.provider.connection)];
                         case 7:
                             if (!(_d.sent())) {
-                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, userOutputTokenAccount, publicKey, swapPoolConfig.getTokenFromSymbol(swapOutCustodyConfig.symbol).mintKey));
+                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, userOutputTokenAccount, publicKey, swapPoolConfig.getTokenFromSymbol(swapOutCustodyConfig.symbol).mintKey, swapPoolConfig.getTokenFromSymbol(swapOutCustodyConfig.symbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID));
                             }
                             marketAccount = positionPoolConfig.getMarketPk(targetCustodyConfig.custodyAccount, collateralCustodyConfig.custodyAccount, side);
                             positionAccount = positionPoolConfig.getPositionFromMarketPk(publicKey, marketAccount);
@@ -2735,7 +2741,7 @@ var PerpetualsClient = (function () {
             if (closeUsersWSOLATA === void 0) { closeUsersWSOLATA = false; }
             if (ephemeralSignerPubkey === void 0) { ephemeralSignerPubkey = undefined; }
             return __awaiter(_this, void 0, void 0, function () {
-                var publicKey, userReceivingTokenAccount, wrappedSolAccount, preInstructions, instructions, postInstructions, additionalSigners, lamports, _a, collateralCustodyConfig, targetCustodyConfig, marketAccount, positionAccount, instruction, closeWsolATAIns, error_2;
+                var publicKey, userReceivingTokenAccount, wrappedSolAccount, preInstructions, instructions, postInstructions, additionalSigners, lamports, _a, collateralCustodyConfig, targetCustodyConfig, marketAccount, positionAccount, rebateMintAccount, instruction, closeWsolATAIns, error_2;
                 return __generator(this, function (_b) {
                     switch (_b.label) {
                         case 0:
@@ -2769,7 +2775,7 @@ var PerpetualsClient = (function () {
                             ];
                             return [3, 5];
                         case 2:
-                            userReceivingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.getTokenFromSymbol(collateralSymbol).mintKey, publicKey, true);
+                            userReceivingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.getTokenFromSymbol(collateralSymbol).mintKey, publicKey, true, poolConfig.getTokenFromSymbol(collateralSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             _a = createUserATA;
                             if (!_a) return [3, 4];
                             return [4, (0, utils_1.checkIfAccountExists)(userReceivingTokenAccount, this.provider.connection)];
@@ -2778,7 +2784,7 @@ var PerpetualsClient = (function () {
                             _b.label = 4;
                         case 4:
                             if (_a) {
-                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, userReceivingTokenAccount, publicKey, poolConfig.getTokenFromSymbol(collateralSymbol).mintKey));
+                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, userReceivingTokenAccount, publicKey, poolConfig.getTokenFromSymbol(collateralSymbol).mintKey, poolConfig.getTokenFromSymbol(collateralSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID));
                             }
                             _b.label = 5;
                         case 5:
@@ -2786,6 +2792,11 @@ var PerpetualsClient = (function () {
                             targetCustodyConfig = poolConfig.custodies.find(function (i) { return i.mintKey.equals(poolConfig.getTokenFromSymbol(marketSymbol).mintKey); });
                             marketAccount = poolConfig.getMarketPk(targetCustodyConfig.custodyAccount, collateralCustodyConfig.custodyAccount, side);
                             positionAccount = poolConfig.getPositionFromMarketPk(publicKey, marketAccount);
+                            rebateMintAccount = {
+                                pubkey: collateralCustodyConfig.mintKey,
+                                isSigner: false,
+                                isWritable: false
+                            };
                             return [4, this.program.methods
                                     .closePosition({
                                     priceWithSlippage: priceWithSlippage,
@@ -2808,9 +2819,10 @@ var PerpetualsClient = (function () {
                                     eventAuthority: this.eventAuthority.publicKey,
                                     tokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
                                     program: this.programId,
-                                    ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY
+                                    ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY,
+                                    receivingMint: collateralCustodyConfig.mintKey
                                 })
-                                    .remainingAccounts(__spreadArray([], (0, getNftAccounts_1.getNftAccounts)(nftTradingAccount, nftReferralAccount, nftRebateTokenAccount, privilege), true))
+                                    .remainingAccounts(__spreadArray(__spreadArray([], (0, getNftAccounts_1.getNftAccounts)(nftTradingAccount, nftReferralAccount, nftRebateTokenAccount, privilege), true), [rebateMintAccount], false))
                                     .instruction()];
                         case 6:
                             instruction = _b.sent();
@@ -2875,22 +2887,21 @@ var PerpetualsClient = (function () {
                                 (0, spl_token_1.createCloseAccountInstruction)((ephemeralSignerPubkey ? ephemeralSignerPubkey : wrappedSolAccount.publicKey), publicKey, publicKey),
                             ];
                             additionalSigners.push(wrappedSolAccount);
-                            return [3, 4];
-                        case 1: return [4, (0, spl_token_1.getAssociatedTokenAddress)(swapPoolConfig.getTokenFromSymbol(outputSymbol).mintKey, publicKey, true)];
-                        case 2:
-                            userReceivingTokenAccount = _b.sent();
+                            return [3, 3];
+                        case 1:
+                            userReceivingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(swapPoolConfig.getTokenFromSymbol(outputSymbol).mintKey, publicKey, true, swapPoolConfig.getTokenFromSymbol(outputSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             return [4, (0, utils_1.checkIfAccountExists)(userReceivingTokenAccount, this.provider.connection)];
-                        case 3:
+                        case 2:
                             if (!(_b.sent())) {
-                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, userReceivingTokenAccount, publicKey, swapPoolConfig.getTokenFromSymbol(outputSymbol).mintKey));
+                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, userReceivingTokenAccount, publicKey, swapPoolConfig.getTokenFromSymbol(outputSymbol).mintKey, swapPoolConfig.getTokenFromSymbol(outputSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID));
                             }
-                            _b.label = 4;
-                        case 4:
-                            userCollateralTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(positionPoolConfig.getTokenFromSymbol(collateralSymbol).mintKey, publicKey, true);
+                            _b.label = 3;
+                        case 3:
+                            userCollateralTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(positionPoolConfig.getTokenFromSymbol(collateralSymbol).mintKey, publicKey, true, positionPoolConfig.getTokenFromSymbol(collateralSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             return [4, (0, utils_1.checkIfAccountExists)(userCollateralTokenAccount, this.provider.connection)];
-                        case 5:
+                        case 4:
                             if (!(_b.sent())) {
-                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, userCollateralTokenAccount, publicKey, positionPoolConfig.getTokenFromSymbol(collateralSymbol).mintKey));
+                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, userCollateralTokenAccount, publicKey, positionPoolConfig.getTokenFromSymbol(collateralSymbol).mintKey, positionPoolConfig.getTokenFromSymbol(collateralSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID));
                             }
                             marketAccount = positionPoolConfig.getMarketPk(targetCustodyConfig.custodyAccount, collateralCustodyConfig.custodyAccount, side);
                             positionAccount = positionPoolConfig.getPositionFromMarketPk(publicKey, marketAccount);
@@ -2939,11 +2950,11 @@ var PerpetualsClient = (function () {
                                     collateralCustodyTokenAccount: collateralCustodyConfig.tokenAccount,
                                     eventAuthority: this.eventAuthority.publicKey,
                                     tokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
-                                    ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY
+                                    ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY,
                                 })
                                     .remainingAccounts(__spreadArray(__spreadArray(__spreadArray([], (0, getNftAccounts_1.getNftAccounts)(nftTradingAccount, nftReferralAccount, nftRebateTokenAccount, privilege), true), custodyAccountMetas, true), custodyOracleAccountMetas, true))
                                     .instruction()];
-                        case 6:
+                        case 5:
                             instruction = _b.sent();
                             instructions.push(instruction);
                             return [2, {
@@ -2961,7 +2972,7 @@ var PerpetualsClient = (function () {
             if (skipBalanceChecks === void 0) { skipBalanceChecks = false; }
             if (ephemeralSignerPubkey === void 0) { ephemeralSignerPubkey = undefined; }
             return __awaiter(_this, void 0, void 0, function () {
-                var publicKey, userInputCustodyConfig, collateralCustodyConfig, targetCustodyConfig, marketAccount, positionAccount, wrappedSolAccount, preInstructions, instructions, postInstructions, additionalSigners, userInputTokenAccount, lamports, unWrappedSolBalance, _a, tokenAccountBalance, _b, userOutputTokenAccount, inx, err_2;
+                var publicKey, userInputCustodyConfig, collateralCustodyConfig, targetCustodyConfig, marketAccount, positionAccount, wrappedSolAccount, preInstructions, instructions, postInstructions, additionalSigners, userInputTokenAccount, lamports, unWrappedSolBalance, _a, tokenAccountBalance, _b, userOutputTokenAccount, rebateMintAccount, inx, err_2;
                 return __generator(this, function (_c) {
                     switch (_c.label) {
                         case 0:
@@ -3019,7 +3030,7 @@ var PerpetualsClient = (function () {
                             ];
                             return [3, 6];
                         case 3:
-                            userInputTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.getTokenFromSymbol(userInputTokenSymbol).mintKey, publicKey, true);
+                            userInputTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.getTokenFromSymbol(userInputTokenSymbol).mintKey, publicKey, true, poolConfig.getTokenFromSymbol(userInputTokenSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             if (!!skipBalanceChecks) return [3, 6];
                             return [4, (0, utils_1.checkIfAccountExists)(userInputTokenAccount, this.provider.connection)];
                         case 4:
@@ -3035,12 +3046,17 @@ var PerpetualsClient = (function () {
                             }
                             _c.label = 6;
                         case 6:
-                            userOutputTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.getTokenFromSymbol(targetCustodyConfig.symbol).mintKey, publicKey, true);
+                            userOutputTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.getTokenFromSymbol(targetCustodyConfig.symbol).mintKey, publicKey, true, poolConfig.getTokenFromSymbol(targetCustodyConfig.symbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             return [4, (0, utils_1.checkIfAccountExists)(userOutputTokenAccount, this.provider.connection)];
                         case 7:
                             if (!(_c.sent())) {
-                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, userOutputTokenAccount, publicKey, poolConfig.getTokenFromSymbol(targetCustodyConfig.symbol).mintKey));
+                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, userOutputTokenAccount, publicKey, poolConfig.getTokenFromSymbol(targetCustodyConfig.symbol).mintKey, poolConfig.getTokenFromSymbol(targetCustodyConfig.symbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID));
                             }
+                            rebateMintAccount = {
+                                pubkey: collateralCustodyConfig.mintKey,
+                                isSigner: false,
+                                isWritable: false
+                            };
                             _c.label = 8;
                         case 8:
                             _c.trys.push([8, 10, , 11]);
@@ -3073,9 +3089,11 @@ var PerpetualsClient = (function () {
                                     tokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
                                     eventAuthority: this.eventAuthority.publicKey,
                                     program: this.programId,
-                                    ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY
+                                    ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY,
+                                    fundingMint: userInputCustodyConfig.mintKey,
+                                    fundingTokenProgram: spl_token_1.TOKEN_PROGRAM_ID
                                 })
-                                    .remainingAccounts(__spreadArray([], (0, getNftAccounts_1.getNftAccounts)(nftTradingAccount, nftReferralAccount, nftRebateTokenAccount, privilege), true))
+                                    .remainingAccounts(__spreadArray(__spreadArray([], (0, getNftAccounts_1.getNftAccounts)(nftTradingAccount, nftReferralAccount, nftRebateTokenAccount, privilege), true), [rebateMintAccount], false))
                                     .instruction()];
                         case 9:
                             inx = _c.sent();
@@ -3099,7 +3117,7 @@ var PerpetualsClient = (function () {
             if (nftRebateTokenAccount === void 0) { nftRebateTokenAccount = web3_js_1.PublicKey.default; }
             if (ephemeralSignerPubkey === void 0) { ephemeralSignerPubkey = undefined; }
             return __awaiter(_this, void 0, void 0, function () {
-                var publicKey, userOutputCustodyConfig, collateralCustodyConfig, targetCustodyConfig, marketAccount, positionAccount, wrappedSolAccount, preInstructions, instructions, postInstructions, additionalSigners, userReceivingTokenAccount, lamports, userCollateralTokenAccount, inx, err_3;
+                var publicKey, userOutputCustodyConfig, collateralCustodyConfig, targetCustodyConfig, marketAccount, positionAccount, wrappedSolAccount, preInstructions, instructions, postInstructions, additionalSigners, userReceivingTokenAccount, lamports, userCollateralTokenAccount, rebateMintAccount, inx, err_3;
                 return __generator(this, function (_a) {
                     switch (_a.label) {
                         case 0:
@@ -3146,26 +3164,30 @@ var PerpetualsClient = (function () {
                                 (0, spl_token_1.createCloseAccountInstruction)((ephemeralSignerPubkey ? ephemeralSignerPubkey : wrappedSolAccount.publicKey), publicKey, publicKey),
                             ];
                             additionalSigners.push(wrappedSolAccount);
-                            return [3, 4];
-                        case 1: return [4, (0, spl_token_1.getAssociatedTokenAddress)(poolConfig.getTokenFromSymbol(userOutputTokenSymbol).mintKey, publicKey, true)];
-                        case 2:
-                            userReceivingTokenAccount = _a.sent();
+                            return [3, 3];
+                        case 1:
+                            userReceivingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.getTokenFromSymbol(userOutputTokenSymbol).mintKey, publicKey, true, poolConfig.getTokenFromSymbol(userOutputTokenSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             return [4, (0, utils_1.checkIfAccountExists)(userReceivingTokenAccount, this.provider.connection)];
-                        case 3:
+                        case 2:
                             if (!(_a.sent())) {
-                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, userReceivingTokenAccount, publicKey, poolConfig.getTokenFromSymbol(userOutputTokenSymbol).mintKey));
+                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, userReceivingTokenAccount, publicKey, poolConfig.getTokenFromSymbol(userOutputTokenSymbol).mintKey, poolConfig.getTokenFromSymbol(userOutputTokenSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID));
                             }
-                            _a.label = 4;
-                        case 4:
-                            userCollateralTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.getTokenFromSymbol(collateralTokenSymbol).mintKey, publicKey, true);
+                            _a.label = 3;
+                        case 3:
+                            userCollateralTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.getTokenFromSymbol(collateralTokenSymbol).mintKey, publicKey, true, poolConfig.getTokenFromSymbol(userOutputTokenSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             return [4, (0, utils_1.checkIfAccountExists)(userCollateralTokenAccount, this.provider.connection)];
-                        case 5:
+                        case 4:
                             if (!(_a.sent())) {
-                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, userCollateralTokenAccount, publicKey, poolConfig.getTokenFromSymbol(collateralTokenSymbol).mintKey));
+                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, userCollateralTokenAccount, publicKey, poolConfig.getTokenFromSymbol(collateralTokenSymbol).mintKey, poolConfig.getTokenFromSymbol(userOutputTokenSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID));
                             }
-                            _a.label = 6;
-                        case 6:
-                            _a.trys.push([6, 8, , 9]);
+                            rebateMintAccount = {
+                                pubkey: collateralCustodyConfig.mintKey,
+                                isSigner: false,
+                                isWritable: false
+                            };
+                            _a.label = 5;
+                        case 5:
+                            _a.trys.push([5, 7, , 8]);
                             return [4, this.program.methods
                                     .closeAndSwap({
                                     priceWithSlippage: priceWithSlippage,
@@ -3190,22 +3212,25 @@ var PerpetualsClient = (function () {
                                     dispensingCustody: userOutputCustodyConfig.custodyAccount,
                                     dispensingOracleAccount: this.useExtOracleAccount ? userOutputCustodyConfig.extOracleAccount : userOutputCustodyConfig.intOracleAccount,
                                     dispensingCustodyTokenAccount: userOutputCustodyConfig.tokenAccount,
-                                    tokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
                                     eventAuthority: this.eventAuthority.publicKey,
                                     program: this.programId,
-                                    ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY
+                                    ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY,
+                                    receivingMint: userOutputCustodyConfig.mintKey,
+                                    receivingTokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
+                                    collateralMint: collateralCustodyConfig.mintKey,
+                                    collateralTokenProgram: spl_token_1.TOKEN_PROGRAM_ID
                                 })
-                                    .remainingAccounts(__spreadArray([], (0, getNftAccounts_1.getNftAccounts)(nftTradingAccount, nftReferralAccount, nftRebateTokenAccount, privilege), true))
+                                    .remainingAccounts(__spreadArray(__spreadArray([], (0, getNftAccounts_1.getNftAccounts)(nftTradingAccount, nftReferralAccount, nftRebateTokenAccount, privilege), true), [rebateMintAccount], false))
                                     .instruction()];
-                        case 7:
+                        case 6:
                             inx = _a.sent();
                             instructions.push(inx);
-                            return [3, 9];
-                        case 8:
+                            return [3, 8];
+                        case 7:
                             err_3 = _a.sent();
                             console.error("perpClient CloseAndSwap error:: ", err_3);
                             throw err_3;
-                        case 9: return [2, {
+                        case 8: return [2, {
                                 instructions: __spreadArray(__spreadArray(__spreadArray([], preInstructions, true), instructions, true), postInstructions, true),
                                 additionalSigners: additionalSigners
                             }];
@@ -3237,26 +3262,24 @@ var PerpetualsClient = (function () {
                             instructions = [];
                             postInstructions = [];
                             additionalSigners = [];
-                            if (!(userInputTokenSymbol == 'SOL' && userOutputTokenSymbol == 'WSOL')) return [3, 5];
-                            return [4, (0, spl_token_1.getAssociatedTokenAddress)(spl_token_1.NATIVE_MINT, publicKey, true)];
-                        case 1:
-                            wsolAssociatedTokenAccount = _f.sent();
+                            if (!(userInputTokenSymbol == 'SOL' && userOutputTokenSymbol == 'WSOL')) return [3, 4];
+                            wsolAssociatedTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(spl_token_1.NATIVE_MINT, publicKey, true);
                             return [4, (0, utils_1.checkIfAccountExists)(wsolAssociatedTokenAccount, this.provider.connection)];
-                        case 2:
+                        case 1:
                             wsolATAExist = _f.sent();
                             if (!wsolATAExist) {
                                 instructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, wsolAssociatedTokenAccount, publicKey, spl_token_1.NATIVE_MINT));
                             }
-                            if (!!skipBalanceChecks) return [3, 4];
+                            if (!!skipBalanceChecks) return [3, 3];
                             _a = anchor_1.BN.bind;
                             return [4, this.provider.connection.getBalance(publicKey)];
-                        case 3:
+                        case 2:
                             unWrappedSolBalance = new (_a.apply(anchor_1.BN, [void 0, _f.sent()]))();
                             if (unWrappedSolBalance.lt(amountIn)) {
                                 throw "Insufficient SOL Funds";
                             }
-                            _f.label = 4;
-                        case 4:
+                            _f.label = 3;
+                        case 3:
                             instructions.push(web3_js_1.SystemProgram.transfer({
                                 fromPubkey: publicKey,
                                 toPubkey: wsolAssociatedTokenAccount,
@@ -3266,7 +3289,7 @@ var PerpetualsClient = (function () {
                                     instructions: __spreadArray(__spreadArray(__spreadArray([], preInstructions, true), instructions, true), postInstructions, true),
                                     additionalSigners: additionalSigners
                                 }];
-                        case 5:
+                        case 4:
                             if (userInputTokenSymbol == 'WSOL' && userOutputTokenSymbol == 'SOL') {
                                 console.log("WSOL=> SOL : NOTE : ONLY WAY IS TO CLOSE THE WSOL ATA and GET ALL SOL ");
                                 wsolAssociatedTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(spl_token_1.NATIVE_MINT, publicKey, true);
@@ -3277,19 +3300,19 @@ var PerpetualsClient = (function () {
                                         additionalSigners: additionalSigners
                                     }];
                             }
-                            _f.label = 6;
-                        case 6:
-                            _f.trys.push([6, 19, , 20]);
-                            if (!(userInputTokenSymbol == 'SOL')) return [3, 9];
+                            _f.label = 5;
+                        case 5:
+                            _f.trys.push([5, 17, , 18]);
+                            if (!(userInputTokenSymbol == 'SOL')) return [3, 8];
                             console.log("userInputTokenSymbol === sol", userInputTokenSymbol);
                             return [4, (0, spl_token_1.getMinimumBalanceForRentExemptAccount)(this.provider.connection)];
-                        case 7:
+                        case 6:
                             accCreationLamports = (_f.sent());
                             console.log("accCreationLamports:", accCreationLamports);
                             lamports = amountIn.add(new anchor_1.BN(accCreationLamports));
                             _b = anchor_1.BN.bind;
                             return [4, this.provider.connection.getBalance(publicKey)];
-                        case 8:
+                        case 7:
                             unWrappedSolBalance = new (_b.apply(anchor_1.BN, [void 0, _f.sent()]))();
                             if (unWrappedSolBalance.lt(amountIn)) {
                                 throw "Insufficient SOL Funds";
@@ -3312,25 +3335,25 @@ var PerpetualsClient = (function () {
                             postInstructions = [
                                 (0, spl_token_1.createCloseAccountInstruction)((ephemeralSignerPubkey ? ephemeralSignerPubkey : wrappedSolAccount.publicKey), publicKey, publicKey),
                             ];
-                            return [3, 12];
-                        case 9:
-                            userInputTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.getTokenFromSymbol(userInputTokenSymbol).mintKey, publicKey, true);
+                            return [3, 11];
+                        case 8:
+                            userInputTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.getTokenFromSymbol(userInputTokenSymbol).mintKey, publicKey, true, poolConfig.getTokenFromSymbol(userOutputTokenSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             return [4, (0, utils_1.checkIfAccountExists)(userInputTokenAccount, this.provider.connection)];
-                        case 10:
+                        case 9:
                             if (!(_f.sent())) {
                                 throw "Insufficient Funds , Token Account doesn't exist";
                             }
-                            if (!!skipBalanceChecks) return [3, 12];
+                            if (!!skipBalanceChecks) return [3, 11];
                             _c = anchor_1.BN.bind;
                             return [4, this.provider.connection.getTokenAccountBalance(userInputTokenAccount)];
-                        case 11:
+                        case 10:
                             tokenAccountBalance = new (_c.apply(anchor_1.BN, [void 0, (_f.sent()).value.amount]))();
                             if (tokenAccountBalance.lt(amountIn)) {
                                 throw "Insufficient Funds need more ".concat(amountIn.sub(tokenAccountBalance), " tokens");
                             }
-                            _f.label = 12;
-                        case 12:
-                            if (!(userOutputTokenSymbol == 'SOL')) return [3, 13];
+                            _f.label = 11;
+                        case 11:
+                            if (!(userOutputTokenSymbol == 'SOL')) return [3, 12];
                             lamports = (this.minimumBalanceForRentExemptAccountLamports);
                             if (!ephemeralSignerPubkey) {
                                 wrappedSolAccount = new web3_js_1.Keypair();
@@ -3350,22 +3373,21 @@ var PerpetualsClient = (function () {
                             postInstructions = [
                                 (0, spl_token_1.createCloseAccountInstruction)((ephemeralSignerPubkey ? ephemeralSignerPubkey : wrappedSolAccount.publicKey), publicKey, publicKey),
                             ];
-                            return [3, 17];
-                        case 13: return [4, (0, spl_token_1.getAssociatedTokenAddress)(poolConfig.getTokenFromSymbol(userOutputTokenSymbol).mintKey, publicKey, true)];
-                        case 14:
-                            userOutputTokenAccount = _f.sent();
+                            return [3, 15];
+                        case 12:
+                            userOutputTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.getTokenFromSymbol(userOutputTokenSymbol).mintKey, publicKey, true, poolConfig.getTokenFromSymbol(userOutputTokenSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             _d = createUserATA;
-                            if (!_d) return [3, 16];
+                            if (!_d) return [3, 14];
                             return [4, (0, utils_1.checkIfAccountExists)(userOutputTokenAccount, this.provider.connection)];
-                        case 15:
+                        case 13:
                             _d = !(_f.sent());
-                            _f.label = 16;
-                        case 16:
+                            _f.label = 14;
+                        case 14:
                             if (_d) {
-                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, userOutputTokenAccount, publicKey, poolConfig.getTokenFromSymbol(userOutputTokenSymbol).mintKey));
+                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, userOutputTokenAccount, publicKey, poolConfig.getTokenFromSymbol(userOutputTokenSymbol).mintKey, poolConfig.getTokenFromSymbol(userOutputTokenSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID));
                             }
-                            _f.label = 17;
-                        case 17:
+                            _f.label = 15;
+                        case 15:
                             custodyAccountMetas = [];
                             custodyOracleAccountMetas = [];
                             for (_i = 0, _e = poolConfig.custodies; _i < _e.length; _i++) {
@@ -3402,25 +3424,28 @@ var PerpetualsClient = (function () {
                                     dispensingCustodyOracleAccount: this.useExtOracleAccount ? userOutputCustodyConfig.extOracleAccount : userOutputCustodyConfig.intOracleAccount,
                                     dispensingCustodyTokenAccount: userOutputCustodyConfig.tokenAccount,
                                     eventAuthority: this.eventAuthority.publicKey,
-                                    tokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
                                     program: this.programId,
-                                    ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY
+                                    ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY,
+                                    fundingMint: userInputCustodyConfig.mintKey,
+                                    fundingTokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
+                                    receivingMint: userOutputCustodyConfig.mintKey,
+                                    receivingTokenProgram: spl_token_1.TOKEN_PROGRAM_ID
                                 })
                                     .remainingAccounts(__spreadArray(__spreadArray([], custodyAccountMetas, true), custodyOracleAccountMetas, true))
                                     .instruction()];
-                        case 18:
+                        case 16:
                             inx = _f.sent();
                             instructions.push(inx);
                             if (userOutputTokenSymbol == 'SOL' && unWrapSol) {
                                 closeWsolATAIns = (0, spl_token_1.createCloseAccountInstruction)(userOutputTokenAccount, publicKey, publicKey);
                                 instructions.push(closeWsolATAIns);
                             }
-                            return [3, 20];
-                        case 19:
+                            return [3, 18];
+                        case 17:
                             err_4 = _f.sent();
                             console.error("perpClient Swap error:: ", err_4);
                             throw err_4;
-                        case 20: return [2, {
+                        case 18: return [2, {
                                 instructions: __spreadArray(__spreadArray(__spreadArray([], preInstructions, true), instructions, true), postInstructions, true),
                                 additionalSigners: additionalSigners
                             }];
@@ -3514,7 +3539,7 @@ var PerpetualsClient = (function () {
                             if (!collateralCustodyConfig || !targetCustodyConfig) {
                                 throw "payTokenCustody not found";
                             }
-                            userPayingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(collateralCustodyConfig.mintKey, publicKey, true);
+                            userPayingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(collateralCustodyConfig.mintKey, publicKey, true, poolConfig.getTokenFromSymbol(collateralSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             preInstructions = [];
                             instructions = [];
                             postInstructions = [];
@@ -3584,6 +3609,7 @@ var PerpetualsClient = (function () {
                                 tokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
                                 program: this.programId,
                                 ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY,
+                                fundingMint: collateralCustodyConfig.mintKey
                             })
                                 .instruction()];
                         case 7:
@@ -3653,7 +3679,7 @@ var PerpetualsClient = (function () {
                             ];
                             return [3, 6];
                         case 3:
-                            userInputTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(inputCustodyConfig.mintKey, publicKey, true);
+                            userInputTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(inputCustodyConfig.mintKey, publicKey, true, poolConfigSwap.getTokenFromSymbol(inputSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             if (!!skipBalanceChecks) return [3, 6];
                             return [4, (0, utils_1.checkIfAccountExists)(userInputTokenAccount, this.provider.connection)];
                         case 4:
@@ -3669,11 +3695,11 @@ var PerpetualsClient = (function () {
                             }
                             _d.label = 6;
                         case 6:
-                            userCollateralTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(collateralCustodyConfig.mintKey, publicKey, true);
+                            userCollateralTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(collateralCustodyConfig.mintKey, publicKey, true, poolConfigPosition.getTokenFromSymbol(collateralSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             return [4, (0, utils_1.checkIfAccountExists)(userCollateralTokenAccount, this.provider.connection)];
                         case 7:
                             if (!(_d.sent())) {
-                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, userCollateralTokenAccount, publicKey, collateralCustodyConfig.mintKey));
+                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, userCollateralTokenAccount, publicKey, collateralCustodyConfig.mintKey, poolConfigPosition.getTokenFromSymbol(collateralSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID));
                             }
                             custodyAccountMetas = [];
                             custodyOracleAccountMetas = [];
@@ -3782,7 +3808,7 @@ var PerpetualsClient = (function () {
                             ];
                             return [3, 5];
                         case 2:
-                            userReceivingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.getTokenFromSymbol(collateralSymbol).mintKey, publicKey, true);
+                            userReceivingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.getTokenFromSymbol(collateralSymbol).mintKey, publicKey, true, poolConfig.getTokenFromSymbol(collateralSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             _a = createUserATA;
                             if (!_a) return [3, 4];
                             return [4, (0, utils_1.checkIfAccountExists)(userReceivingTokenAccount, this.provider.connection)];
@@ -3791,7 +3817,7 @@ var PerpetualsClient = (function () {
                             _b.label = 4;
                         case 4:
                             if (_a) {
-                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, userReceivingTokenAccount, publicKey, poolConfig.getTokenFromSymbol(collateralSymbol).mintKey));
+                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, userReceivingTokenAccount, publicKey, poolConfig.getTokenFromSymbol(collateralSymbol).mintKey, poolConfig.getTokenFromSymbol(collateralSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID));
                             }
                             _b.label = 5;
                         case 5:
@@ -3816,7 +3842,8 @@ var PerpetualsClient = (function () {
                                     eventAuthority: this.eventAuthority.publicKey,
                                     tokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
                                     program: this.programId,
-                                    ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY
+                                    ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY,
+                                    receivingMint: collateralCustodyConfig.mintKey
                                 })
                                     .instruction()];
                         case 6:
@@ -3880,19 +3907,19 @@ var PerpetualsClient = (function () {
                             ];
                             return [3, 3];
                         case 1:
-                            userReceivingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfigSwap.getTokenFromSymbol(outputSymbol).mintKey, publicKey, true);
+                            userReceivingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfigSwap.getTokenFromSymbol(outputSymbol).mintKey, publicKey, true, poolConfigSwap.getTokenFromSymbol(outputSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             return [4, (0, utils_1.checkIfAccountExists)(userReceivingTokenAccount, this.provider.connection)];
                         case 2:
                             if (!(_b.sent())) {
-                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, userReceivingTokenAccount, publicKey, poolConfigSwap.getTokenFromSymbol(outputSymbol).mintKey));
+                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, userReceivingTokenAccount, publicKey, poolConfigSwap.getTokenFromSymbol(outputSymbol).mintKey, poolConfigSwap.getTokenFromSymbol(outputSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID));
                             }
                             _b.label = 3;
                         case 3:
-                            userCollateralTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfigPosition.getTokenFromSymbol(collateralSymbol).mintKey, publicKey, true);
+                            userCollateralTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfigPosition.getTokenFromSymbol(collateralSymbol).mintKey, publicKey, true, poolConfigPosition.getTokenFromSymbol(collateralSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             return [4, (0, utils_1.checkIfAccountExists)(userCollateralTokenAccount, this.provider.connection)];
                         case 4:
                             if (!(_b.sent())) {
-                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, userCollateralTokenAccount, publicKey, poolConfigPosition.getTokenFromSymbol(collateralSymbol).mintKey));
+                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, userCollateralTokenAccount, publicKey, poolConfigPosition.getTokenFromSymbol(collateralSymbol).mintKey, poolConfigPosition.getTokenFromSymbol(collateralSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID));
                             }
                             marketAccount = poolConfigPosition.getMarketPk(targetCustodyConfig.custodyAccount, collateralCustodyConfig.custodyAccount, side);
                             positionAccount = poolConfigPosition.getPositionFromMarketPk(publicKey, marketAccount);
@@ -3960,7 +3987,7 @@ var PerpetualsClient = (function () {
             if (nftReferralAccount === void 0) { nftReferralAccount = web3_js_1.PublicKey.default; }
             if (nftRebateTokenAccount === void 0) { nftRebateTokenAccount = web3_js_1.PublicKey.default; }
             return __awaiter(_this, void 0, void 0, function () {
-                var publicKey, collateralCustodyConfig, targetCustodyConfig, marketAccount, preInstructions, instructions, postInstructions, additionalSigners, instruction;
+                var publicKey, collateralCustodyConfig, targetCustodyConfig, marketAccount, preInstructions, instructions, postInstructions, additionalSigners, rebateMintAccount, instruction;
                 return __generator(this, function (_a) {
                     switch (_a.label) {
                         case 0:
@@ -3979,6 +4006,11 @@ var PerpetualsClient = (function () {
                             instructions = [];
                             postInstructions = [];
                             additionalSigners = [];
+                            rebateMintAccount = {
+                                pubkey: collateralCustodyConfig.mintKey,
+                                isSigner: false,
+                                isWritable: false
+                            };
                             return [4, this.program.methods
                                     .increaseSize({
                                     priceWithSlippage: priceWithSlippage,
@@ -4002,7 +4034,7 @@ var PerpetualsClient = (function () {
                                     program: this.programId,
                                     ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY
                                 })
-                                    .remainingAccounts(__spreadArray([], (0, getNftAccounts_1.getNftAccounts)(nftTradingAccount, nftReferralAccount, nftRebateTokenAccount, privilege), true))
+                                    .remainingAccounts(__spreadArray(__spreadArray([], (0, getNftAccounts_1.getNftAccounts)(nftTradingAccount, nftReferralAccount, nftRebateTokenAccount, privilege), true), [rebateMintAccount], false))
                                     .instruction()];
                         case 1:
                             instruction = _a.sent();
@@ -4020,7 +4052,7 @@ var PerpetualsClient = (function () {
             if (nftReferralAccount === void 0) { nftReferralAccount = web3_js_1.PublicKey.default; }
             if (nftRebateTokenAccount === void 0) { nftRebateTokenAccount = web3_js_1.PublicKey.default; }
             return __awaiter(_this, void 0, void 0, function () {
-                var publicKey, collateralCustodyConfig, targetCustodyConfig, marketAccount, preInstructions, instructions, postInstructions, additionalSigners, instruction;
+                var publicKey, collateralCustodyConfig, targetCustodyConfig, marketAccount, preInstructions, instructions, postInstructions, additionalSigners, rebateMintAccount, instruction;
                 return __generator(this, function (_a) {
                     switch (_a.label) {
                         case 0:
@@ -4039,6 +4071,11 @@ var PerpetualsClient = (function () {
                             instructions = [];
                             postInstructions = [];
                             additionalSigners = [];
+                            rebateMintAccount = {
+                                pubkey: collateralCustodyConfig.mintKey,
+                                isSigner: false,
+                                isWritable: false
+                            };
                             return [4, this.program.methods
                                     .decreaseSize({
                                     priceWithSlippage: priceWithSlippage,
@@ -4062,7 +4099,7 @@ var PerpetualsClient = (function () {
                                     program: this.programId,
                                     ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY
                                 })
-                                    .remainingAccounts(__spreadArray([], (0, getNftAccounts_1.getNftAccounts)(nftTradingAccount, nftReferralAccount, nftRebateTokenAccount, privilege), true))
+                                    .remainingAccounts(__spreadArray(__spreadArray([], (0, getNftAccounts_1.getNftAccounts)(nftTradingAccount, nftReferralAccount, nftRebateTokenAccount, privilege), true), [rebateMintAccount], false))
                                     .instruction()];
                         case 1:
                             instruction = _a.sent();
@@ -4095,7 +4132,7 @@ var PerpetualsClient = (function () {
                             _f.label = 1;
                         case 1:
                             _f.trys.push([1, 10, , 11]);
-                            userPayingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(payTokenCustodyConfig.mintKey, publicKey, true);
+                            userPayingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(payTokenCustodyConfig.mintKey, publicKey, true, poolConfig.getTokenFromSymbol(payTokenSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             lpTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.stakedLpTokenMint, publicKey, true);
                             custodyAccountMetas = [];
                             custodyOracleAccountMetas = [];
@@ -4191,7 +4228,9 @@ var PerpetualsClient = (function () {
                                 eventAuthority: this.eventAuthority.publicKey,
                                 tokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
                                 program: this.programId,
-                                ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY
+                                ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY,
+                                fundingMint: payTokenCustodyConfig.mintKey,
+                                fundingTokenProgram: spl_token_1.TOKEN_PROGRAM_ID
                             })
                                 .remainingAccounts(__spreadArray(__spreadArray(__spreadArray([], custodyAccountMetas, true), custodyOracleAccountMetas, true), markets, true))
                                 .instruction()];
@@ -4266,7 +4305,7 @@ var PerpetualsClient = (function () {
                             ];
                             return [3, 7];
                         case 4:
-                            userInputTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(inputCustodyConfig.mintKey, publicKey, true);
+                            userInputTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(inputCustodyConfig.mintKey, publicKey, true, poolConfig.getTokenFromSymbol(inputSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             if (!!skipBalanceChecks) return [3, 7];
                             return [4, (0, utils_1.checkIfAccountExists)(userInputTokenAccount, this.provider.connection)];
                         case 5:
@@ -4326,7 +4365,9 @@ var PerpetualsClient = (function () {
                                     tokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
                                     eventAuthority: this.eventAuthority.publicKey,
                                     program: this.programId,
-                                    ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY
+                                    ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY,
+                                    fundingMint: inputCustodyConfig.mintKey,
+                                    fundingTokenProgram: spl_token_1.TOKEN_PROGRAM_ID
                                 })
                                     .remainingAccounts(__spreadArray(__spreadArray(__spreadArray([], custodyAccountMetas, true), custodyOracleAccountMetas, true), markets, true))
                                     .instruction()];
@@ -4409,7 +4450,7 @@ var PerpetualsClient = (function () {
                             ];
                             return [3, 5];
                         case 2:
-                            userReceivingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.getTokenFromSymbol(recieveTokenSymbol).mintKey, publicKey, true);
+                            userReceivingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.getTokenFromSymbol(recieveTokenSymbol).mintKey, publicKey, true, poolConfig.getTokenFromSymbol(recieveTokenSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             _d = createUserATA;
                             if (!_d) return [3, 4];
                             return [4, (0, utils_1.checkIfAccountExists)(userReceivingTokenAccount, this.provider.connection)];
@@ -4418,7 +4459,7 @@ var PerpetualsClient = (function () {
                             _e.label = 4;
                         case 4:
                             if (_d) {
-                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, userReceivingTokenAccount, publicKey, poolConfig.getTokenFromSymbol(recieveTokenSymbol).mintKey));
+                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, userReceivingTokenAccount, publicKey, poolConfig.getTokenFromSymbol(recieveTokenSymbol).mintKey, poolConfig.getTokenFromSymbol(recieveTokenSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID));
                             }
                             _e.label = 5;
                         case 5: return [4, this.program.methods
@@ -4440,7 +4481,9 @@ var PerpetualsClient = (function () {
                                 eventAuthority: this.eventAuthority.publicKey,
                                 tokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
                                 program: this.programId,
-                                ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY
+                                ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY,
+                                receivingMint: recieveTokenCustodyConfig.mintKey,
+                                receivingTokenProgram: spl_token_1.TOKEN_PROGRAM_ID
                             })
                                 .remainingAccounts(__spreadArray(__spreadArray(__spreadArray([], custodyAccountMetas, true), custodyOracleAccountMetas, true), markets, true))
                                 .instruction()];
@@ -4567,7 +4610,7 @@ var PerpetualsClient = (function () {
                         additionalSigners = [];
                         _a.label = 1;
                     case 1:
-                        _a.trys.push([1, 4, , 5]);
+                        _a.trys.push([1, 3, , 4]);
                         nftTradingAccount = web3_js_1.PublicKey.findProgramAddressSync([
                             Buffer.from("trading"),
                             nftMint.toBuffer(),
@@ -4576,9 +4619,7 @@ var PerpetualsClient = (function () {
                             Buffer.from("referral"),
                             publicKey.toBuffer(),
                         ], this.programId)[0];
-                        return [4, (0, spl_token_1.getAssociatedTokenAddress)(nftMint, publicKey, true)];
-                    case 2:
-                        nftTokenAccount = _a.sent();
+                        nftTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(nftMint, publicKey, true);
                         flpStakeAccountMetas = [];
                         for (_i = 0, flpStakeAccounts_1 = flpStakeAccounts; _i < flpStakeAccounts_1.length; _i++) {
                             flpStakeAccountPk = flpStakeAccounts_1[_i];
@@ -4601,15 +4642,15 @@ var PerpetualsClient = (function () {
                                 tradingAccount: nftTradingAccount
                             })
                                 .instruction()];
-                    case 3:
+                    case 2:
                         updateNftTradingAccountInstruction = _a.sent();
                         instructions.push(updateNftTradingAccountInstruction);
-                        return [3, 5];
-                    case 4:
+                        return [3, 4];
+                    case 3:
                         err_10 = _a.sent();
                         console.log("perpClient updateNftAccount error:: ", err_10);
                         throw err_10;
-                    case 5: return [2, {
+                    case 4: return [2, {
                             instructions: __spreadArray(__spreadArray(__spreadArray([], preInstructions, true), instructions, true), postInstructions, true),
                             additionalSigners: additionalSigners
                         }];
@@ -4782,13 +4823,11 @@ var PerpetualsClient = (function () {
                         additionalSigners = [];
                         _a.label = 1;
                     case 1:
-                        _a.trys.push([1, 4, , 5]);
+                        _a.trys.push([1, 3, , 4]);
                         lpTokenMint = poolConfig.stakedLpTokenMint;
                         poolStakedLpVault = web3_js_1.PublicKey.findProgramAddressSync([Buffer.from("staked_lp_token_account"), poolConfig.poolAddress.toBuffer(), lpTokenMint.toBuffer()], this.programId)[0];
                         flpStakeAccount = web3_js_1.PublicKey.findProgramAddressSync([Buffer.from("stake"), owner.toBuffer(), poolConfig.poolAddress.toBuffer()], this.programId)[0];
-                        return [4, (0, spl_token_1.getAssociatedTokenAddress)(poolConfig.stakedLpTokenMint, owner, true)];
-                    case 2:
-                        userLpTokenAccount = _a.sent();
+                        userLpTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.stakedLpTokenMint, owner, true);
                         return [4, this.program.methods
                                 .depositStake({
                                 depositAmount: depositAmount
@@ -4805,18 +4844,19 @@ var PerpetualsClient = (function () {
                                 systemProgram: web3_js_1.SystemProgram.programId,
                                 tokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
                                 eventAuthority: this.eventAuthority.publicKey,
-                                program: this.programId
+                                program: this.programId,
+                                lpTokenMint: poolConfig.stakedLpTokenMint
                             })
                                 .instruction()];
-                    case 3:
+                    case 2:
                         depositStakeInstruction = _a.sent();
                         instructions.push(depositStakeInstruction);
-                        return [3, 5];
-                    case 4:
+                        return [3, 4];
+                    case 3:
                         err_14 = _a.sent();
                         console.log("perpClient depositStaking error:: ", err_14);
                         throw err_14;
-                    case 5: return [2, {
+                    case 4: return [2, {
                             instructions: __spreadArray(__spreadArray(__spreadArray([], preInstructions, true), instructions, true), postInstructions, true),
                             additionalSigners: additionalSigners
                         }];
@@ -5103,6 +5143,7 @@ var PerpetualsClient = (function () {
                                     tokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
                                     eventAuthority: this.eventAuthority.publicKey,
                                     program: this.program.programId,
+                                    lpMint: poolConfig.stakedLpTokenMint
                                 })
                                     .instruction()];
                         case 4:
@@ -5140,7 +5181,7 @@ var PerpetualsClient = (function () {
                             _b.trys.push([1, 5, , 6]);
                             pool = poolConfig.poolAddress;
                             flpStakeAccount = web3_js_1.PublicKey.findProgramAddressSync([Buffer.from("stake"), publicKey.toBuffer(), pool.toBuffer()], this.program.programId)[0];
-                            receivingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(rewardCustodyMint, publicKey, true);
+                            receivingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(rewardCustodyMint, publicKey, true, poolConfig.getTokenFromSymbol(rewardSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             _a = createUserATA;
                             if (!_a) return [3, 3];
                             return [4, (0, utils_1.checkIfAccountExists)(receivingTokenAccount, this.provider.connection)];
@@ -5149,7 +5190,7 @@ var PerpetualsClient = (function () {
                             _b.label = 3;
                         case 3:
                             if (_a) {
-                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, receivingTokenAccount, publicKey, rewardCustodyMint));
+                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, receivingTokenAccount, publicKey, rewardCustodyMint, poolConfig.getTokenFromSymbol(rewardSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID));
                             }
                             tradingAccount = [];
                             if (nftTradingAccount) {
@@ -5182,7 +5223,8 @@ var PerpetualsClient = (function () {
                                     systemProgram: web3_js_1.SystemProgram.programId,
                                     tokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
                                     eventAuthority: this.eventAuthority.publicKey,
-                                    ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY
+                                    ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY,
+                                    receivingMint: rewardCustodyMint
                                 })
                                     .remainingAccounts(__spreadArray(__spreadArray([], tradingAccount, true), boostingAccount, true))
                                     .instruction()];
@@ -5251,18 +5293,19 @@ var PerpetualsClient = (function () {
             });
         }); };
         this.distributeReward = function (rewardAmount, rewardSymbol, poolConfig) { return __awaiter(_this, void 0, void 0, function () {
-            var publicKey, rewardCustodyMint, instructions, additionalSigners, fundingAccount, rewardVault, rewardTokenAccount, distributeReward, err_22;
+            var publicKey, rewardToken, rewardCustodyMint, instructions, additionalSigners, fundingAccount, rewardVault, rewardTokenAccount, distributeReward, err_22;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         publicKey = this.provider.wallet.publicKey;
-                        rewardCustodyMint = poolConfig.getTokenFromSymbol(rewardSymbol).mintKey;
+                        rewardToken = poolConfig.getTokenFromSymbol(rewardSymbol);
+                        rewardCustodyMint = rewardToken.mintKey;
                         instructions = [];
                         additionalSigners = [];
                         _a.label = 1;
                     case 1:
                         _a.trys.push([1, 3, , 4]);
-                        fundingAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(rewardCustodyMint, publicKey, true);
+                        fundingAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(rewardCustodyMint, publicKey, true, rewardToken.isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                         rewardVault = web3_js_1.PublicKey.findProgramAddressSync([Buffer.from("reward_vault")], this.programFbnftReward.programId)[0];
                         rewardTokenAccount = web3_js_1.PublicKey.findProgramAddressSync([Buffer.from("reward_token_account")], this.programFbnftReward.programId)[0];
                         return [4, this.programFbnftReward.methods
@@ -5295,12 +5338,13 @@ var PerpetualsClient = (function () {
         this.collectNftReward = function (rewardSymbol, poolConfig, nftMint, createUserATA) {
             if (createUserATA === void 0) { createUserATA = true; }
             return __awaiter(_this, void 0, void 0, function () {
-                var publicKey, rewardCustodyMint, instructions, additionalSigners, nftTokenAccount, metadataAccount, receivingTokenAccount, _a, rewardRecord, rewardVault, rewardTokenAccount, nftTransferAuthority, collectNftReward, err_23;
+                var publicKey, rewardToken, rewardCustodyMint, instructions, additionalSigners, nftTokenAccount, metadataAccount, receivingTokenAccount, _a, rewardRecord, rewardVault, rewardTokenAccount, nftTransferAuthority, collectNftReward, err_23;
                 return __generator(this, function (_b) {
                     switch (_b.label) {
                         case 0:
                             publicKey = this.provider.wallet.publicKey;
-                            rewardCustodyMint = poolConfig.getTokenFromSymbol(rewardSymbol).mintKey;
+                            rewardToken = poolConfig.getTokenFromSymbol(rewardSymbol);
+                            rewardCustodyMint = rewardToken.mintKey;
                             instructions = [];
                             additionalSigners = [];
                             _b.label = 1;
@@ -5308,7 +5352,7 @@ var PerpetualsClient = (function () {
                             _b.trys.push([1, 5, , 6]);
                             nftTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(nftMint, publicKey, true);
                             metadataAccount = web3_js_1.PublicKey.findProgramAddressSync([Buffer.from("metadata"), constants_1.METAPLEX_PROGRAM_ID.toBuffer(), nftMint.toBuffer()], constants_1.METAPLEX_PROGRAM_ID)[0];
-                            receivingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(rewardCustodyMint, publicKey, true);
+                            receivingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(rewardCustodyMint, publicKey, true, rewardToken.isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             _a = createUserATA;
                             if (!_a) return [3, 3];
                             return [4, (0, utils_1.checkIfAccountExists)(receivingTokenAccount, this.provider.connection)];
@@ -5317,7 +5361,7 @@ var PerpetualsClient = (function () {
                             _b.label = 3;
                         case 3:
                             if (_a) {
-                                instructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, receivingTokenAccount, publicKey, rewardCustodyMint));
+                                instructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, receivingTokenAccount, publicKey, rewardCustodyMint, rewardToken.isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID));
                             }
                             rewardRecord = web3_js_1.PublicKey.findProgramAddressSync([Buffer.from("reward_record"), nftMint.toBuffer()], this.programFbnftReward.programId)[0];
                             rewardVault = web3_js_1.PublicKey.findProgramAddressSync([Buffer.from("reward_vault")], this.programFbnftReward.programId)[0];
@@ -5358,13 +5402,14 @@ var PerpetualsClient = (function () {
         this.collectAndDistributeFee = function (rewardSymbol, poolConfig, createUserATA, nftTradingAccount) {
             if (createUserATA === void 0) { createUserATA = true; }
             return __awaiter(_this, void 0, void 0, function () {
-                var publicKey, rewardCustodyMint, rewardCustodyConfig, preInstructions, instructions, postInstructions, additionalSigners, pool, flpStakeAccount, receivingTokenAccount, _a, tradingAccount, rewardVault, rewardTokenAccount, withdrawStakeInstruction, err_24;
+                var publicKey, rewardToken, rewardCustodyMint, rewardCustodyConfig, preInstructions, instructions, postInstructions, additionalSigners, pool, flpStakeAccount, receivingTokenAccount, _a, tradingAccount, rewardVault, rewardTokenAccount, withdrawStakeInstruction, err_24;
                 return __generator(this, function (_b) {
                     switch (_b.label) {
                         case 0:
                             publicKey = this.provider.wallet.publicKey;
-                            rewardCustodyMint = poolConfig.getTokenFromSymbol(rewardSymbol).mintKey;
-                            rewardCustodyConfig = poolConfig.custodies.find(function (i) { return i.mintKey.equals(poolConfig.getTokenFromSymbol(rewardSymbol).mintKey); });
+                            rewardToken = poolConfig.getTokenFromSymbol(rewardSymbol);
+                            rewardCustodyMint = rewardToken.mintKey;
+                            rewardCustodyConfig = poolConfig.custodies.find(function (i) { return i.mintKey.equals(rewardToken.mintKey); });
                             preInstructions = [];
                             instructions = [];
                             postInstructions = [];
@@ -5374,7 +5419,7 @@ var PerpetualsClient = (function () {
                             _b.trys.push([1, 5, , 6]);
                             pool = poolConfig.poolAddress;
                             flpStakeAccount = web3_js_1.PublicKey.findProgramAddressSync([Buffer.from("stake"), publicKey.toBuffer(), pool.toBuffer()], this.program.programId)[0];
-                            receivingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(rewardCustodyMint, publicKey, true);
+                            receivingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(rewardCustodyMint, publicKey, true, rewardToken.isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             _a = createUserATA;
                             if (!_a) return [3, 3];
                             return [4, (0, utils_1.checkIfAccountExists)(receivingTokenAccount, this.provider.connection)];
@@ -5383,7 +5428,7 @@ var PerpetualsClient = (function () {
                             _b.label = 3;
                         case 3:
                             if (_a) {
-                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, receivingTokenAccount, publicKey, rewardCustodyMint));
+                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, receivingTokenAccount, publicKey, rewardCustodyMint, rewardToken.isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID));
                             }
                             tradingAccount = [];
                             if (nftTradingAccount) {
@@ -5502,19 +5547,17 @@ var PerpetualsClient = (function () {
                             additionalSigners = [];
                             _b.label = 1;
                         case 1:
-                            _b.trys.push([1, 6, , 7]);
-                            return [4, (0, spl_token_1.getAssociatedTokenAddress)(poolConfig.getTokenFromSymbol(collateralSymbol).mintKey, positionAccount.owner)];
-                        case 2:
-                            userReceivingTokenAccount = _b.sent();
+                            _b.trys.push([1, 5, , 6]);
+                            userReceivingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.getTokenFromSymbol(collateralSymbol).mintKey, positionAccount.owner, false, poolConfig.getTokenFromSymbol(collateralSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             _a = createUserATA;
-                            if (!_a) return [3, 4];
+                            if (!_a) return [3, 3];
                             return [4, (0, utils_1.checkIfAccountExists)(userReceivingTokenAccount, this.provider.connection)];
-                        case 3:
+                        case 2:
                             _a = !(_b.sent());
-                            _b.label = 4;
-                        case 4:
+                            _b.label = 3;
+                        case 3:
                             if (_a) {
-                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(payerPubkey, userReceivingTokenAccount, positionAccount.owner, poolConfig.getTokenFromSymbol(collateralSymbol).mintKey));
+                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(payerPubkey, userReceivingTokenAccount, positionAccount.owner, poolConfig.getTokenFromSymbol(collateralSymbol).mintKey, poolConfig.getTokenFromSymbol(collateralSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID));
                             }
                             return [4, this.program.methods
                                     .forceClosePosition({
@@ -5537,22 +5580,23 @@ var PerpetualsClient = (function () {
                                     tokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
                                     eventAuthority: this.eventAuthority.publicKey,
                                     program: this.programId,
-                                    ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY
+                                    ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY,
+                                    receivingMint: collateralCustodyConfig.mintKey
                                 })
                                     .instruction()];
-                        case 5:
+                        case 4:
                             forceClosePosition = _b.sent();
                             instructions.push(forceClosePosition);
                             if (collateralSymbol == 'WSOL' && closeUsersWSOLATA) {
                                 closeWsolATAIns = (0, spl_token_1.createCloseAccountInstruction)(userReceivingTokenAccount, positionAccount.owner, positionAccount.owner);
                                 postInstructions.push(closeWsolATAIns);
                             }
-                            return [3, 7];
-                        case 6:
+                            return [3, 6];
+                        case 5:
                             err_26 = _b.sent();
                             console.log("perpClient forceClosePosition error:: ", err_26);
                             throw err_26;
-                        case 7: return [2, {
+                        case 6: return [2, {
                                 instructions: __spreadArray(__spreadArray(__spreadArray([], preInstructions, true), instructions, true), postInstructions, true),
                                 additionalSigners: additionalSigners
                             }];
@@ -5573,30 +5617,28 @@ var PerpetualsClient = (function () {
                             reserveCustodyConfig = poolConfig.custodies.find(function (i) { return i.mintKey.equals(poolConfig.getTokenFromSymbol(reserveSymbol).mintKey); });
                             collateralCustodyConfig = poolConfig.custodies.find(function (i) { return i.mintKey.equals(poolConfig.getTokenFromSymbol(collateralSymbol).mintKey); });
                             marketAccount = poolConfig.getMarketPk(targetCustodyConfig.custodyAccount, collateralCustodyConfig.custodyAccount, side);
-                            return [4, (0, spl_token_1.getAssociatedTokenAddress)(poolConfig.getTokenFromSymbol(reserveSymbol).mintKey, publicKey)];
-                        case 1:
-                            userReserveTokenAccount = _c.sent();
+                            userReserveTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.getTokenFromSymbol(reserveSymbol).mintKey, publicKey, false, poolConfig.getTokenFromSymbol(reserveSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             preInstructions = [];
                             instructions = [];
                             postInstructions = [];
                             additionalSigners = [];
-                            _c.label = 2;
-                        case 2:
-                            _c.trys.push([2, 10, , 11]);
-                            if (!(reserveSymbol == 'SOL')) return [3, 5];
+                            _c.label = 1;
+                        case 1:
+                            _c.trys.push([1, 9, , 10]);
+                            if (!(reserveSymbol == 'SOL')) return [3, 4];
                             console.log("reserveSymbol === SOL", reserveSymbol);
                             accCreationLamports = this.minimumBalanceForRentExemptAccountLamports;
                             lamports = reserveAmount.add(new anchor_1.BN(accCreationLamports));
-                            if (!!skipBalanceChecks) return [3, 4];
+                            if (!!skipBalanceChecks) return [3, 3];
                             _a = anchor_1.BN.bind;
                             return [4, this.provider.connection.getBalance(publicKey)];
-                        case 3:
+                        case 2:
                             unWrappedSolBalance = new (_a.apply(anchor_1.BN, [void 0, _c.sent()]))();
                             if (unWrappedSolBalance.lt(lamports)) {
                                 throw "Insufficient SOL Funds";
                             }
-                            _c.label = 4;
-                        case 4:
+                            _c.label = 3;
+                        case 3:
                             if (!ephemeralSignerPubkey) {
                                 wrappedSolAccount = new web3_js_1.Keypair();
                                 additionalSigners.push(wrappedSolAccount);
@@ -5615,22 +5657,22 @@ var PerpetualsClient = (function () {
                                 (0, spl_token_1.createCloseAccountInstruction)((ephemeralSignerPubkey ? ephemeralSignerPubkey : wrappedSolAccount.publicKey), publicKey, publicKey),
                             ];
                             additionalSigners.push(wrappedSolAccount);
-                            return [3, 8];
-                        case 5: return [4, (0, utils_1.checkIfAccountExists)(userReserveTokenAccount, this.provider.connection)];
-                        case 6:
+                            return [3, 7];
+                        case 4: return [4, (0, utils_1.checkIfAccountExists)(userReserveTokenAccount, this.provider.connection)];
+                        case 5:
                             if (!(_c.sent())) {
                                 throw "Insufficient Funds , token Account doesn't exist";
                             }
-                            if (!!skipBalanceChecks) return [3, 8];
+                            if (!!skipBalanceChecks) return [3, 7];
                             _b = anchor_1.BN.bind;
                             return [4, this.provider.connection.getTokenAccountBalance(userReserveTokenAccount)];
-                        case 7:
+                        case 6:
                             tokenAccountBalance = new (_b.apply(anchor_1.BN, [void 0, (_c.sent()).value.amount]))();
                             if (tokenAccountBalance.lt(reserveAmount)) {
                                 throw "Insufficient Funds need more ".concat(reserveAmount.sub(tokenAccountBalance), " tokens");
                             }
-                            _c.label = 8;
-                        case 8:
+                            _c.label = 7;
+                        case 7:
                             positionAccount = poolConfig.getPositionFromMarketPk(publicKey, marketAccount);
                             orderAccount = poolConfig.getOrderFromMarketPk(publicKey, marketAccount);
                             return [4, this.program.methods
@@ -5660,18 +5702,19 @@ var PerpetualsClient = (function () {
                                     tokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
                                     eventAuthority: this.eventAuthority.publicKey,
                                     program: this.programId,
-                                    ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY
+                                    ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY,
+                                    fundingMint: reserveCustodyConfig.mintKey
                                 })
                                     .instruction()];
-                        case 9:
+                        case 8:
                             placeLimitOrder = _c.sent();
                             instructions.push(placeLimitOrder);
-                            return [3, 11];
-                        case 10:
+                            return [3, 10];
+                        case 9:
                             err_27 = _c.sent();
                             console.log("perpClient placeLimitOrder error:: ", err_27);
                             throw err_27;
-                        case 11: return [2, {
+                        case 10: return [2, {
                                 instructions: __spreadArray(__spreadArray(__spreadArray([], preInstructions, true), instructions, true), postInstructions, true),
                                 additionalSigners: additionalSigners
                             }];
@@ -5720,7 +5763,7 @@ var PerpetualsClient = (function () {
                             ];
                             return [3, 5];
                         case 2:
-                            userReceivingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.getTokenFromSymbol(reserveSymbol).mintKey, publicKey, true);
+                            userReceivingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.getTokenFromSymbol(reserveSymbol).mintKey, publicKey, true, poolConfig.getTokenFromSymbol(reserveSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             _a = createUserATA;
                             if (!_a) return [3, 4];
                             return [4, (0, utils_1.checkIfAccountExists)(userReceivingTokenAccount, this.provider.connection)];
@@ -5729,7 +5772,7 @@ var PerpetualsClient = (function () {
                             _b.label = 4;
                         case 4:
                             if (_a) {
-                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, userReceivingTokenAccount, publicKey, poolConfig.getTokenFromSymbol(reserveSymbol).mintKey));
+                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, userReceivingTokenAccount, publicKey, poolConfig.getTokenFromSymbol(reserveSymbol).mintKey, poolConfig.getTokenFromSymbol(reserveSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID));
                             }
                             _b.label = 5;
                         case 5:
@@ -5763,6 +5806,7 @@ var PerpetualsClient = (function () {
                                     eventAuthority: this.eventAuthority.publicKey,
                                     program: this.programId,
                                     ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY,
+                                    receivingMint: reserveCustodyConfig.mintKey
                                 })
                                     .instruction()];
                         case 6:
@@ -6104,7 +6148,7 @@ var PerpetualsClient = (function () {
                             if (!false) return [3, 2];
                             return [3, 7];
                         case 2:
-                            userReceivingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.getTokenFromSymbol(receivingSymbol).mintKey, owner, true);
+                            userReceivingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.getTokenFromSymbol(receivingSymbol).mintKey, owner, true, poolConfig.getTokenFromSymbol(receivingSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             _a = createUserATA;
                             if (!_a) return [3, 4];
                             return [4, (0, utils_1.checkIfAccountExists)(userReceivingTokenAccount, this.provider.connection)];
@@ -6113,9 +6157,9 @@ var PerpetualsClient = (function () {
                             _d.label = 4;
                         case 4:
                             if (_a) {
-                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(payerPubkey, userReceivingTokenAccount, owner, poolConfig.getTokenFromSymbol(receivingSymbol).mintKey));
+                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(payerPubkey, userReceivingTokenAccount, owner, poolConfig.getTokenFromSymbol(receivingSymbol).mintKey, poolConfig.getTokenFromSymbol(receivingSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID));
                             }
-                            userReceivingTokenAccountCollateral = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.getTokenFromSymbol(collateralSymbol).mintKey, owner, true);
+                            userReceivingTokenAccountCollateral = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.getTokenFromSymbol(collateralSymbol).mintKey, owner, true, poolConfig.getTokenFromSymbol(collateralSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             _b = createUserATA;
                             if (!_b) return [3, 6];
                             return [4, (0, utils_1.checkIfAccountExists)(userReceivingTokenAccountCollateral, this.provider.connection)];
@@ -6124,7 +6168,7 @@ var PerpetualsClient = (function () {
                             _d.label = 6;
                         case 6:
                             if (_b) {
-                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(payerPubkey, userReceivingTokenAccountCollateral, owner, poolConfig.getTokenFromSymbol(collateralSymbol).mintKey));
+                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(payerPubkey, userReceivingTokenAccountCollateral, owner, poolConfig.getTokenFromSymbol(collateralSymbol).mintKey, poolConfig.getTokenFromSymbol(collateralSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID));
                             }
                             _d.label = 7;
                         case 7:
@@ -6168,10 +6212,13 @@ var PerpetualsClient = (function () {
                                     dispensingCustody: receivingCustodyConfig.custodyAccount,
                                     dispensingOracleAccount: this.useExtOracleAccount ? receivingCustodyConfig.extOracleAccount : receivingCustodyConfig.intOracleAccount,
                                     dispensingCustodyTokenAccount: receivingCustodyConfig.tokenAccount,
-                                    tokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
                                     eventAuthority: this.eventAuthority.publicKey,
                                     program: this.programId,
-                                    ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY
+                                    ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY,
+                                    receivingMint: receivingCustodyConfig.mintKey,
+                                    receivingTokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
+                                    collateralMint: collateralCustodyConfig.mintKey,
+                                    collateralTokenProgram: spl_token_1.TOKEN_PROGRAM_ID
                                 })
                                     .remainingAccounts(__spreadArray(__spreadArray([], custodyAccountMetas, true), custodyOracleAccountMetas, true))
                                     .instruction()];
@@ -6213,7 +6260,7 @@ var PerpetualsClient = (function () {
                             if (!false) return [3, 2];
                             return [3, 5];
                         case 2:
-                            userReceivingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.getTokenFromSymbol(collateralSymbol).mintKey, owner, true);
+                            userReceivingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.getTokenFromSymbol(collateralSymbol).mintKey, owner, true, poolConfig.getTokenFromSymbol(collateralSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             _a = createUserATA;
                             if (!_a) return [3, 4];
                             return [4, (0, utils_1.checkIfAccountExists)(userReceivingTokenAccount, this.provider.connection)];
@@ -6222,7 +6269,7 @@ var PerpetualsClient = (function () {
                             _b.label = 4;
                         case 4:
                             if (_a) {
-                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(payerPubkey, userReceivingTokenAccount, owner, poolConfig.getTokenFromSymbol(collateralSymbol).mintKey));
+                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(payerPubkey, userReceivingTokenAccount, owner, poolConfig.getTokenFromSymbol(collateralSymbol).mintKey, poolConfig.getTokenFromSymbol(collateralSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID));
                             }
                             _b.label = 5;
                         case 5:
@@ -6250,7 +6297,8 @@ var PerpetualsClient = (function () {
                                     tokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
                                     eventAuthority: this.eventAuthority.publicKey,
                                     program: this.programId,
-                                    ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY
+                                    ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY,
+                                    receivingMint: collateralCustodyConfig.mintKey
                                 })
                                     .instruction()];
                         case 6:
@@ -6354,14 +6402,12 @@ var PerpetualsClient = (function () {
                     case 0:
                         publicKey = this.provider.wallet.publicKey;
                         custodyConfig = poolConfig.custodies.find(function (i) { return i.mintKey.equals(poolConfig.getTokenFromSymbol(rewardSymbol).mintKey); });
-                        return [4, (0, spl_token_1.getAssociatedTokenAddress)(poolConfig.getTokenFromSymbol(rewardSymbol).mintKey, publicKey, true)];
-                    case 1:
-                        receivingTokenAccount = _a.sent();
+                        receivingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.getTokenFromSymbol(rewardSymbol).mintKey, publicKey, true, poolConfig.getTokenFromSymbol(rewardSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                         instructions = [];
                         additionalSigners = [];
-                        _a.label = 2;
-                    case 2:
-                        _a.trys.push([2, 4, , 5]);
+                        _a.label = 1;
+                    case 1:
+                        _a.trys.push([1, 3, , 4]);
                         return [4, this.program.methods
                                 .withdrawFees({})
                                 .accounts({
@@ -6373,18 +6419,19 @@ var PerpetualsClient = (function () {
                                 custody: custodyConfig.custodyAccount,
                                 custodyTokenAccount: custodyConfig.tokenAccount,
                                 receivingTokenAccount: receivingTokenAccount,
-                                tokenProgram: spl_token_1.TOKEN_PROGRAM_ID
+                                tokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
+                                receivingMint: custodyConfig.mintKey
                             })
                                 .instruction()];
-                    case 3:
+                    case 2:
                         withdrawFeesIx = _a.sent();
                         instructions.push(withdrawFeesIx);
-                        return [3, 5];
-                    case 4:
+                        return [3, 4];
+                    case 3:
                         err_38 = _a.sent();
                         console.log("perpClient setPool error:: ", err_38);
                         throw err_38;
-                    case 5: return [2, {
+                    case 4: return [2, {
                             instructions: __spreadArray([], instructions, true),
                             additionalSigners: additionalSigners
                         }];
@@ -6430,10 +6477,9 @@ var PerpetualsClient = (function () {
             });
         }); };
         this.reimburse = function (tokenMint, amountIn, poolConfig) { return __awaiter(_this, void 0, void 0, function () {
-            var custodyAccountMetas, custodyOracleAccountMetas, markets, _i, _a, custody, _b, _c, market, instructions, additionalSigners, custodyConfig, reimburse, _d, _e, err_40;
-            var _f;
-            return __generator(this, function (_g) {
-                switch (_g.label) {
+            var custodyAccountMetas, custodyOracleAccountMetas, markets, _i, _a, custody, _b, _c, market, instructions, additionalSigners, custodyConfig, reimburse, err_40;
+            return __generator(this, function (_d) {
+                switch (_d.label) {
                     case 0:
                         custodyAccountMetas = [];
                         custodyOracleAccountMetas = [];
@@ -6462,38 +6508,36 @@ var PerpetualsClient = (function () {
                         instructions = [];
                         additionalSigners = [];
                         custodyConfig = poolConfig.custodies.find(function (i) { return i.mintKey.equals(tokenMint); });
-                        _g.label = 1;
+                        _d.label = 1;
                     case 1:
-                        _g.trys.push([1, 4, , 5]);
-                        _e = (_d = this.program.methods
-                            .reimburse({ amountIn: amountIn }))
-                            .accounts;
-                        _f = {
-                            admin: this.provider.wallet.publicKey,
-                            multisig: poolConfig.multisig
-                        };
-                        return [4, (0, spl_token_1.getAssociatedTokenAddress)(tokenMint, this.provider.wallet.publicKey, true)];
-                    case 2: return [4, _e.apply(_d, [(_f.fundingAccount = _g.sent(),
-                                _f.perpetuals = poolConfig.perpetuals,
-                                _f.pool = poolConfig.poolAddress,
-                                _f.custody = custodyConfig.custodyAccount,
-                                _f.custodyOracleAccount = this.useExtOracleAccount ? custodyConfig.extOracleAccount : custodyConfig.intOracleAccount,
-                                _f.custodyTokenAccount = custodyConfig.tokenAccount,
-                                _f.tokenProgram = spl_token_1.TOKEN_PROGRAM_ID,
-                                _f.program = poolConfig.programId,
-                                _f.ixSysvar = web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY,
-                                _f)])
-                            .remainingAccounts(__spreadArray(__spreadArray(__spreadArray([], custodyAccountMetas, true), custodyOracleAccountMetas, true), markets, true))
-                            .instruction()];
-                    case 3:
-                        reimburse = _g.sent();
+                        _d.trys.push([1, 3, , 4]);
+                        return [4, this.program.methods
+                                .reimburse({ amountIn: amountIn })
+                                .accounts({
+                                admin: this.provider.wallet.publicKey,
+                                multisig: poolConfig.multisig,
+                                fundingAccount: (0, spl_token_1.getAssociatedTokenAddressSync)(tokenMint, this.provider.wallet.publicKey, true),
+                                perpetuals: poolConfig.perpetuals,
+                                pool: poolConfig.poolAddress,
+                                custody: custodyConfig.custodyAccount,
+                                custodyOracleAccount: this.useExtOracleAccount ? custodyConfig.extOracleAccount : custodyConfig.intOracleAccount,
+                                custodyTokenAccount: custodyConfig.tokenAccount,
+                                tokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
+                                program: poolConfig.programId,
+                                ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY,
+                                fundingMint: custodyConfig.mintKey
+                            })
+                                .remainingAccounts(__spreadArray(__spreadArray(__spreadArray([], custodyAccountMetas, true), custodyOracleAccountMetas, true), markets, true))
+                                .instruction()];
+                    case 2:
+                        reimburse = _d.sent();
                         instructions.push(reimburse);
-                        return [3, 5];
-                    case 4:
-                        err_40 = _g.sent();
+                        return [3, 4];
+                    case 3:
+                        err_40 = _d.sent();
                         console.log("perpClient setPool error:: ", err_40);
                         throw err_40;
-                    case 5: return [2, {
+                    case 4: return [2, {
                             instructions: __spreadArray([], instructions, true),
                             additionalSigners: additionalSigners
                         }];
@@ -6563,7 +6607,7 @@ var PerpetualsClient = (function () {
                             compoundingTokenMint = poolConfig.compoundingTokenMint;
                             lpTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(poolConfig.stakedLpTokenMint, publicKey, true);
                             compoundingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(compoundingTokenMint, publicKey, true);
-                            fundingAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(inCustodyConfig.mintKey, publicKey, true);
+                            fundingAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(inCustodyConfig.mintKey, publicKey, true, poolConfig.getTokenFromSymbol(inTokenSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             custodyAccountMetas = [];
                             custodyOracleAccountMetas = [];
                             markets = [];
@@ -6662,7 +6706,9 @@ var PerpetualsClient = (function () {
                                     tokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
                                     eventAuthority: this.eventAuthority.publicKey,
                                     program: this.program.programId,
-                                    ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY
+                                    ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY,
+                                    fundingMint: inCustodyConfig.mintKey,
+                                    fundingTokenProgram: spl_token_1.TOKEN_PROGRAM_ID
                                 })
                                     .remainingAccounts(__spreadArray(__spreadArray(__spreadArray([], custodyAccountMetas, true), custodyOracleAccountMetas, true), markets, true))
                                     .instruction()];
@@ -6720,7 +6766,7 @@ var PerpetualsClient = (function () {
                             ];
                             return [3, 4];
                         case 1:
-                            userReceivingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(outCustodyConfig.mintKey, publicKey, true);
+                            userReceivingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(outCustodyConfig.mintKey, publicKey, true, poolConfig.getTokenFromSymbol(outTokenSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             _a = createUserATA;
                             if (!_a) return [3, 3];
                             return [4, (0, utils_1.checkIfAccountExists)(userReceivingTokenAccount, this.provider.connection)];
@@ -6729,7 +6775,7 @@ var PerpetualsClient = (function () {
                             _e.label = 3;
                         case 3:
                             if (_a) {
-                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, userReceivingTokenAccount, publicKey, outCustodyConfig.mintKey));
+                                preInstructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, userReceivingTokenAccount, publicKey, outCustodyConfig.mintKey, poolConfig.getTokenFromSymbol(outTokenSymbol).isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID));
                             }
                             _e.label = 4;
                         case 4:
@@ -6784,7 +6830,9 @@ var PerpetualsClient = (function () {
                                     tokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
                                     eventAuthority: this.eventAuthority.publicKey,
                                     program: this.program.programId,
-                                    ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY
+                                    ixSysvar: web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY,
+                                    receivingMint: outCustodyConfig.mintKey,
+                                    receivingTokenProgram: spl_token_1.TOKEN_PROGRAM_ID
                                 })
                                     .remainingAccounts(__spreadArray(__spreadArray(__spreadArray([], custodyAccountMetas, true), custodyOracleAccountMetas, true), markets, true))
                                     .instruction()];
@@ -7182,18 +7230,19 @@ var PerpetualsClient = (function () {
             });
         }); };
         this.rewardDistribution = function (counter, owner, rewardAmount, rewardSymbol, poolConfig) { return __awaiter(_this, void 0, void 0, function () {
-            var publicKey, rewardCustodyMint, instructions, additionalSigners, fundingAccount, rewardVault, rewardTokenAccount, rewardRecord, distributeReward, err_50;
+            var publicKey, rewardToken, rewardCustodyMint, instructions, additionalSigners, fundingAccount, rewardVault, rewardTokenAccount, rewardRecord, distributeReward, err_50;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         publicKey = this.provider.wallet.publicKey;
-                        rewardCustodyMint = poolConfig.getTokenFromSymbol(rewardSymbol).mintKey;
+                        rewardToken = poolConfig.getTokenFromSymbol(rewardSymbol);
+                        rewardCustodyMint = rewardToken.mintKey;
                         instructions = [];
                         additionalSigners = [];
                         _a.label = 1;
                     case 1:
                         _a.trys.push([1, 3, , 4]);
-                        fundingAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(rewardCustodyMint, publicKey, true);
+                        fundingAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(rewardCustodyMint, publicKey, true, rewardToken.isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                         rewardVault = web3_js_1.PublicKey.findProgramAddressSync([Buffer.from("reward_vault")], this.programRewardDistribution.programId)[0];
                         rewardTokenAccount = web3_js_1.PublicKey.findProgramAddressSync([Buffer.from("reward_token_account")], this.programRewardDistribution.programId)[0];
                         rewardRecord = web3_js_1.PublicKey.findProgramAddressSync([Buffer.from("reward_record"), owner.toBuffer(), rewardVault.toBuffer(), counter.toArrayLike(Buffer, 'le', 8)], this.programRewardDistribution.programId)[0];
@@ -7234,18 +7283,19 @@ var PerpetualsClient = (function () {
         this.collectReward = function (counter, owner, rewardSymbol, poolConfig, createUserATA) {
             if (createUserATA === void 0) { createUserATA = true; }
             return __awaiter(_this, void 0, void 0, function () {
-                var publicKey, rewardCustodyMint, instructions, additionalSigners, receivingTokenAccount, _a, rewardVault, rewardTokenAccount, rewardRecord, transferAuthority, collectRaffleReward, err_51;
+                var publicKey, rewardToken, rewardCustodyMint, instructions, additionalSigners, receivingTokenAccount, _a, rewardVault, rewardTokenAccount, rewardRecord, transferAuthority, collectRaffleReward, err_51;
                 return __generator(this, function (_b) {
                     switch (_b.label) {
                         case 0:
                             publicKey = this.provider.wallet.publicKey;
-                            rewardCustodyMint = poolConfig.getTokenFromSymbol(rewardSymbol).mintKey;
+                            rewardToken = poolConfig.getTokenFromSymbol(rewardSymbol);
+                            rewardCustodyMint = rewardToken.mintKey;
                             instructions = [];
                             additionalSigners = [];
                             _b.label = 1;
                         case 1:
                             _b.trys.push([1, 5, , 6]);
-                            receivingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(rewardCustodyMint, publicKey, true);
+                            receivingTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(rewardCustodyMint, publicKey, true, rewardToken.isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID);
                             _a = createUserATA;
                             if (!_a) return [3, 3];
                             return [4, (0, utils_1.checkIfAccountExists)(receivingTokenAccount, this.provider.connection)];
@@ -7254,7 +7304,7 @@ var PerpetualsClient = (function () {
                             _b.label = 3;
                         case 3:
                             if (_a) {
-                                instructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, receivingTokenAccount, publicKey, rewardCustodyMint));
+                                instructions.push((0, spl_token_1.createAssociatedTokenAccountInstruction)(publicKey, receivingTokenAccount, publicKey, rewardCustodyMint, rewardToken.isToken2022 ? spl_token_1.TOKEN_2022_PROGRAM_ID : spl_token_1.TOKEN_PROGRAM_ID));
                             }
                             rewardVault = web3_js_1.PublicKey.findProgramAddressSync([Buffer.from("reward_vault")], this.programRewardDistribution.programId)[0];
                             rewardTokenAccount = web3_js_1.PublicKey.findProgramAddressSync([Buffer.from("reward_token_account")], this.programRewardDistribution.programId)[0];