diff --git a/SDWebImageWebPCoder/Classes/SDImageWebPCoder.h b/SDWebImageWebPCoder/Classes/SDImageWebPCoder.h
index 7729681..4a02ddd 100644
--- a/SDWebImageWebPCoder/Classes/SDImageWebPCoder.h
+++ b/SDWebImageWebPCoder/Classes/SDImageWebPCoder.h
@@ -7,7 +7,7 @@
  */
 
 #import <Foundation/Foundation.h>
-#import <SDWebImage/SDImageCoder.h>
+@import SDWebImage;
 
 /**
  Built in coder that supports WebP and animated WebP
diff --git a/SDWebImageWebPCoder/Classes/UIImage+WebP.h b/SDWebImageWebPCoder/Classes/UIImage+WebP.h
index 45d9d20..b935d2f 100644
--- a/SDWebImageWebPCoder/Classes/UIImage+WebP.h
+++ b/SDWebImageWebPCoder/Classes/UIImage+WebP.h
@@ -6,8 +6,7 @@
  * file that was distributed with this source code.
  */
 
-@import UIKit;
-#import <SDWebImage/SDWebImageCompat.h>
+@import SDWebImage;
 
 // This category is just use as a convenience method. For more detail control, use methods in `UIImage+MultiFormat.h` or directlly use `SDImageCoder`
 @interface UIImage (WebP)