diff --git a/LeadKit/Sources/Protocols/StaticEstimatedViewHeightProtocol.swift b/LeadKit/Sources/Protocols/StaticEstimatedViewHeightProtocol.swift
index 1e9844f4..8ac52b15 100644
--- a/LeadKit/Sources/Protocols/StaticEstimatedViewHeightProtocol.swift
+++ b/LeadKit/Sources/Protocols/StaticEstimatedViewHeightProtocol.swift
@@ -20,7 +20,7 @@
 //  THE SOFTWARE.
 //
 
-import Foundation
+import CoreGraphics
 
 /**
  *  protocol which ensures that specific type can return estimated height of view
diff --git a/LeadKit/Sources/Protocols/StaticViewHeightProtocol.swift b/LeadKit/Sources/Protocols/StaticViewHeightProtocol.swift
index c1e441eb..27bad418 100644
--- a/LeadKit/Sources/Protocols/StaticViewHeightProtocol.swift
+++ b/LeadKit/Sources/Protocols/StaticViewHeightProtocol.swift
@@ -20,7 +20,7 @@
 //  THE SOFTWARE.
 //
 
-import Foundation
+import CoreGraphics
 
 /**
  *  protocol which ensures that specific type can return height of view
diff --git a/LeadKit/Sources/Protocols/ViewHeightProtocol.swift b/LeadKit/Sources/Protocols/ViewHeightProtocol.swift
index 3c29c7bc..f0c92573 100644
--- a/LeadKit/Sources/Protocols/ViewHeightProtocol.swift
+++ b/LeadKit/Sources/Protocols/ViewHeightProtocol.swift
@@ -20,7 +20,7 @@
 //  THE SOFTWARE.
 //
 
-import Foundation
+import CoreGraphics
 
 /**
  *  protocol which ensures that specific type can return height of view for view model