Wallet.updateBalanceX()

[dg314]

I would like to update the X-chain balance of a wallet so that I can access the wallet's balance of various ANT tokens. Since updateBalanceX() is private, the only way to do this seems to be calling updateUtxosX(). However, this function calls updateBalanceX() without awaiting its promise, so I'm not sure how I can tell when it has finished updating the X-chain balance. For my use case, I would prefer if the function did the following:

await this.updateUnknownAssetsX();
await this.updateBalanceX();

I understand that the function is probably written as it currently is because updateBalanceX() is a costly function. Maybe a solution could be to have a boolean parameter waitForBalance for the function updateBalanceX() with the following code:

if (waitForBalance) {
    await this.updateBalanceX();
else {
    this.updateBalanceX();
}

Apologies in advance if there is another solution to this issue that I'm missing.