fixed updateBalance bug