diff --git a/tee-worker/litentry/core/assertion-build/src/dynamic/contracts/token_holding_amount/TokenQueryLogic.sol b/tee-worker/litentry/core/assertion-build/src/dynamic/contracts/token_holding_amount/TokenQueryLogic.sol
index 25b1bf515f..0db17c665c 100644
--- a/tee-worker/litentry/core/assertion-build/src/dynamic/contracts/token_holding_amount/TokenQueryLogic.sol
+++ b/tee-worker/litentry/core/assertion-build/src/dynamic/contracts/token_holding_amount/TokenQueryLogic.sol
@@ -28,7 +28,6 @@ import "../openzeppelin/Strings.sol";
 import "./Constants.sol";
 
 abstract contract TokenQueryLogic is TokenHoldingAmount {
-	mapping(string => mapping(uint32 => string)) tokenAddresses;
 	mapping(string => TokenInfo[]) internal tokenInfo;
 
 	// TODO fix it for erc20 token, same token for different networks has different decimals.
@@ -108,7 +107,7 @@ abstract contract TokenQueryLogic is TokenHoldingAmount {
 							if (
 								Strings.equal(
 									solanaTokenBalance[i].mint,
-									tokenAddresses[tokenName][network]
+									tokenContractAddress
 								)
 							) {
 								(