diff --git a/Source/JTSImageInfo.h b/Source/JTSImageInfo.h
index f65a5ac..7cfbc5a 100644
--- a/Source/JTSImageInfo.h
+++ b/Source/JTSImageInfo.h
@@ -6,7 +6,7 @@
 //  Copyright (c) 2014 Nice Boy LLC. All rights reserved.
 //
 
-@import Foundation;
+@import UIKit;
 
 @interface JTSImageInfo : NSObject
 
diff --git a/Source/JTSSimpleImageDownloader.h b/Source/JTSSimpleImageDownloader.h
index bef844e..76328c6 100644
--- a/Source/JTSSimpleImageDownloader.h
+++ b/Source/JTSSimpleImageDownloader.h
@@ -6,7 +6,7 @@
 //  Copyright (c) 2014 Nice Boy LLC. All rights reserved.
 //
 
-@import Foundation;
+@import UIKit;
 
 @interface JTSSimpleImageDownloader : NSObject