diff --git a/CountryPickerView.podspec b/CountryPickerView.podspec
index 006af6f..1fae69f 100755
--- a/CountryPickerView.podspec
+++ b/CountryPickerView.podspec
@@ -1,7 +1,7 @@
 Pod::Spec.new do |spec|
 
   spec.name         = "CountryPickerView"
-  spec.version      = "3.1.2"
+  spec.version      = "3.1.3"
   spec.summary      = "A simple, customizable view for selecting countries in iOS apps."
   spec.homepage     = "https://github.com/kizitonwose/CountryPickerView"
   spec.license      = "MIT"
diff --git a/CountryPickerView/Delegate+DataSource.swift b/CountryPickerView/Delegate+DataSource.swift
index 50bf44d..076a9ae 100644
--- a/CountryPickerView/Delegate+DataSource.swift
+++ b/CountryPickerView/Delegate+DataSource.swift
@@ -6,7 +6,7 @@
 //  Copyright © 2018 Kizito Nwose. All rights reserved.
 //
 
-import Foundation
+import UIKit
 
 public protocol CountryPickerViewDelegate: class {
     /// Called when the user selects a country from the list.
@@ -81,7 +81,7 @@ public protocol CountryPickerViewDataSource: class {
     func localeForCountryNameInList(in countryPickerView: CountryPickerView) -> Locale
     
     /// An array of countries you wish to exclude from the list of countries.
-    func excludedCountriesInList(in countryPickerView: CountryPickerView) -> [Country]
+    func excludedCountries(in countryPickerView: CountryPickerView) -> [Country]
 }
 
 // MARK:- CountryPickerViewDataSource default implementations
@@ -151,7 +151,7 @@ public extension CountryPickerViewDataSource {
         return Locale.current
     }
     
-    func excludedCountriesInList(in countryPickerView: CountryPickerView) -> [Country] {
+    func excludedCountries(in countryPickerView: CountryPickerView) -> [Country] {
         return []
     }
 }