From 69fdc40f6ac1a9ffab1958a1af44d4e9d6fb62a6 Mon Sep 17 00:00:00 2001
From: Kizito Nwose <kizitonwose@gmail.com>
Date: Sun, 2 Aug 2020 17:35:27 +0200
Subject: [PATCH] Fix logic.

---
 CountryPickerView/CountryPickerView.swift           | 2 +-
 CountryPickerView/CountryPickerViewController.swift | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/CountryPickerView/CountryPickerView.swift b/CountryPickerView/CountryPickerView.swift
index ae38fb0..5a6801b 100644
--- a/CountryPickerView/CountryPickerView.swift
+++ b/CountryPickerView/CountryPickerView.swift
@@ -200,7 +200,7 @@ public class CountryPickerView: NibView {
     }()
     
     internal var usableCountries: [Country] {
-        let excluded = dataSource?.excludedCountriesInList(in: self) ?? []
+        let excluded = dataSource?.excludedCountries(in: self) ?? []
         return countries.filter { return !excluded.contains($0) }
     }
 }
diff --git a/CountryPickerView/CountryPickerViewController.swift b/CountryPickerView/CountryPickerViewController.swift
index 1d02692..10389d2 100644
--- a/CountryPickerView/CountryPickerViewController.swift
+++ b/CountryPickerView/CountryPickerViewController.swift
@@ -353,6 +353,6 @@ class CountryPickerViewDataSourceInternal: CountryPickerViewDataSource {
     }
     
     var excludedCountries: [Country] {
-        return view.dataSource?.excludedCountriesInList(in: view) ?? preferredCountries(in: view)
+        return view.dataSource?.excludedCountries(in: view) ?? excludedCountries(in: view)
     }
 }