diff --git a/prediction_market_agent_tooling/markets/omen/omen.py b/prediction_market_agent_tooling/markets/omen/omen.py
index 3110880c..f482a364 100644
--- a/prediction_market_agent_tooling/markets/omen/omen.py
+++ b/prediction_market_agent_tooling/markets/omen/omen.py
@@ -408,7 +408,7 @@ def redeem_winnings(api_keys: APIKeys) -> None:
 
     @staticmethod
     def get_trade_balance(api_keys: APIKeys, web3: Web3 | None = None) -> xDai:
-        native_token_balance = get_balances(api_keys.public_key, web3=web3).xdai
+        native_token_balance = get_balances(api_keys.bet_from_address, web3=web3).xdai
         return xdai_type(
             wei_to_xdai(
                 KEEPING_ERC20_TOKEN.balanceOf(api_keys.bet_from_address, web3=web3)