diff --git a/nitro/wallet/wallet.nim b/nitro/wallet/wallet.nim index b499e11..2a9c6c7 100644 --- a/nitro/wallet/wallet.nim +++ b/nitro/wallet/wallet.nim @@ -108,9 +108,12 @@ func balance*(wallet: Wallet, func balance*(wallet: Wallet, channel: ChannelId, asset: EthAddress, - address = wallet.address): UInt256 = + address: EthAddress): UInt256 = wallet.balance(channel, asset, address.toDestination) +func balance*(wallet: Wallet, channel: ChannelId, asset: EthAddress): UInt256 = + wallet.balance(channel, asset, wallet.address) + func total(state: State, asset: EthAddress): UInt256 = var total: UInt256 if balances =? state.outcome.balances(asset):