diff --git a/Sources/ButtonBarViewCell.swift b/Sources/ButtonBarViewCell.swift
index 4f608623..e6789b13 100644
--- a/Sources/ButtonBarViewCell.swift
+++ b/Sources/ButtonBarViewCell.swift
@@ -22,7 +22,7 @@
 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 // THE SOFTWARE.
 
-import Foundation
+import UIKit
 
 open class ButtonBarViewCell: UICollectionViewCell {