Fix to get_balance types