diff --git a/TMCache/TMDiskCache.m b/TMCache/TMDiskCache.m
index a15f07d..c084929 100644
--- a/TMCache/TMDiskCache.m
+++ b/TMCache/TMDiskCache.m
@@ -1,4 +1,5 @@
 #import "TMDiskCache.h"
+#import <UIKit/UIKit.h>
 
 #define TMDiskCacheError(error) if (error) { NSLog(@"%@ (%d) ERROR: %@", \
                                     [[NSString stringWithUTF8String:__FILE__] lastPathComponent], \
diff --git a/TMCache/TMMemoryCache.m b/TMCache/TMMemoryCache.m
index 66972cc..07eca87 100644
--- a/TMCache/TMMemoryCache.m
+++ b/TMCache/TMMemoryCache.m
@@ -1,4 +1,5 @@
 #import "TMMemoryCache.h"
+#import <UIKit/UIKit.h>
 
 NSString * const TMMemoryCachePrefix = @"com.tumblr.TMMemoryCache";