diff --git a/YYWebImage/YYImageCache.m b/YYWebImage/YYImageCache.m
index 65b2779..a492dea 100644
--- a/YYWebImage/YYImageCache.m
+++ b/YYWebImage/YYImageCache.m
@@ -10,7 +10,6 @@
 //
 
 #import "YYImageCache.h"
-#import "YYImage.h"
 #import "UIImage+YYWebImage.h"
 
 #if __has_include(<YYImage/YYImage.h>)
diff --git a/YYWebImage/YYWebImageManager.m b/YYWebImage/YYWebImageManager.m
index 784fd8b..3fa044c 100644
--- a/YYWebImage/YYWebImageManager.m
+++ b/YYWebImage/YYWebImageManager.m
@@ -12,9 +12,14 @@
 #import "YYWebImageManager.h"
 #import "YYImageCache.h"
 #import "YYWebImageOperation.h"
-#import "YYImageCoder.h"
 #import <objc/runtime.h>
 
+#if __has_include( <YYImage/YYImageCoder.h>)
+#import <YYImage/YYImageCoder.h>
+#else
+#import "YYImageCoder.h"
+#endif
+
 #define kNetworkIndicatorDelay (1/30.0)