diff --git a/scripts/balance.js b/scripts/balance.js index 52e0772..377f5f6 100644 --- a/scripts/balance.js +++ b/scripts/balance.js @@ -12,8 +12,10 @@ module.exports = async (callback) => { const [account,] = await web3.eth.getAccounts(); - const wrappedBalance1 = await wrapper.balanceOf(account, nftId1); - const tokenBalance1 = await tokenContract1.balanceOf(account); + const whom = process.argv[7] ? process.argv[7] : account; + + const wrappedBalance1 = await wrapper.balanceOf(whom, nftId1); + const tokenBalance1 = await tokenContract1.balanceOf(whom); console.log(`Wrapped: ${wrappedBalance1}`); console.log(`Unwrapped: ${tokenBalance1}`);