diff --git a/AsyncDisplayKit/ASMultiplexImageNode.h b/AsyncDisplayKit/ASMultiplexImageNode.h
index cac569d3c7..5c02ed6048 100644
--- a/AsyncDisplayKit/ASMultiplexImageNode.h
+++ b/AsyncDisplayKit/ASMultiplexImageNode.h
@@ -6,11 +6,12 @@
  * of patent rights can be found in the PATENTS file in the same directory.
  */
 
+#if TARGET_OS_IOS
+
 #import <AsyncDisplayKit/ASImageNode.h>
 #import <AsyncDisplayKit/ASImageProtocols.h>
-#if TARGET_OS_IOS
 #import <Photos/Photos.h>
-#endif
+
 NS_ASSUME_NONNULL_BEGIN
 
 @protocol ASMultiplexImageNodeDelegate;
@@ -265,4 +266,6 @@ didFinishDownloadingImageWithIdentifier:(ASImageIdentifier)imageIdentifier
 @end
 #endif
 
-NS_ASSUME_NONNULL_END
\ No newline at end of file
+NS_ASSUME_NONNULL_END
+
+#endif
diff --git a/AsyncDisplayKit/ASMultiplexImageNode.mm b/AsyncDisplayKit/ASMultiplexImageNode.mm
index fa58bb88f1..ff35ed3aec 100644
--- a/AsyncDisplayKit/ASMultiplexImageNode.mm
+++ b/AsyncDisplayKit/ASMultiplexImageNode.mm
@@ -6,12 +6,12 @@
  * of patent rights can be found in the PATENTS file in the same directory.
  */
 
-#import "ASMultiplexImageNode.h"
 #if TARGET_OS_IOS
+
+#import "ASMultiplexImageNode.h"
 #import <AssetsLibrary/AssetsLibrary.h>
 
 #import <Photos/Photos.h>
-#endif
 #import <libkern/OSAtomic.h>
 
 #import "ASAvailability.h"
@@ -726,4 +726,6 @@ typedef void(^ASMultiplexImageLoadCompletionBlock)(UIImage *image, id imageIdent
 }
 
 @end
-#endif
\ No newline at end of file
+#endif
+
+#endif