diff --git a/setup.py b/setup.py
index 6aacf32..f079730 100644
--- a/setup.py
+++ b/setup.py
@@ -48,7 +48,7 @@ print(f"New BIP84 testnet address: {address} at index {index}")
 # print wallet balance
 wallet.sync(blockchain, None)
 balance = wallet.get_balance()
-print(f"Wallet balance is: {balance}")
+print(f"Wallet balance is: {balance.total}")
 """
 
 rust_ext = RustExtension(