Fix logic.

kizitonwose · Aug 2, 2020 · 69fdc40 · 69fdc40
1 parent d1d0649
commit 69fdc40
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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)
     }
 }