upgrade getBalance