Fix get_balance query