diff --git a/SDWebImage/UIImage+CacheMemoryCost.h b/SDWebImage/UIImage+CacheMemoryCost.h
index 67136d5e..480230e9 100644
--- a/SDWebImage/UIImage+CacheMemoryCost.h
+++ b/SDWebImage/UIImage+CacheMemoryCost.h
@@ -6,7 +6,7 @@
  * file that was distributed with this source code.
  */
 
-#import <UIKit/UIKit.h>
+#import "SDWebImageCompat.h"
 
 @interface UIImage (CacheMemoryCost)
 
diff --git a/SDWebImage/UIImage+CacheMemoryCost.m b/SDWebImage/UIImage+CacheMemoryCost.m
index 48462dd5..be92eb4b 100644
--- a/SDWebImage/UIImage+CacheMemoryCost.m
+++ b/SDWebImage/UIImage+CacheMemoryCost.m
@@ -7,7 +7,6 @@
  */
 
 #import "UIImage+CacheMemoryCost.h"
-#import "SDWebImageCompat.h"
 #import "objc/runtime.h"
 
 FOUNDATION_STATIC_INLINE NSUInteger SDCacheCostForImage(UIImage *image) {