diff --git a/Sources/Swift/FlexLayout.swift b/Sources/Swift/FlexLayout.swift
index 374831b7..69cd8c49 100644
--- a/Sources/Swift/FlexLayout.swift
+++ b/Sources/Swift/FlexLayout.swift
@@ -13,7 +13,7 @@
 // Created by Luc Dion on 2017-06-19.
 
 import UIKit
-#if FLEXLAYOUT_SWIFT_PACKAGE
+#if FLEXLAYOUT_SWIFT_PACKAGE || SWIFT_PACKAGE
 import FlexLayoutYogaKit
 #endif
 
diff --git a/Sources/Swift/Impl/FlexLayout+Enum.swift b/Sources/Swift/Impl/FlexLayout+Enum.swift
index 6969fac9..295ea1e4 100644
--- a/Sources/Swift/Impl/FlexLayout+Enum.swift
+++ b/Sources/Swift/Impl/FlexLayout+Enum.swift
@@ -14,7 +14,7 @@
 
 import UIKit
 
-#if FLEXLAYOUT_SWIFT_PACKAGE
+#if FLEXLAYOUT_SWIFT_PACKAGE || SWIFT_PACKAGE
 import FlexLayoutYoga
 
 extension YGFlexDirection {
diff --git a/Sources/Swift/Impl/FlexLayout+Private.swift b/Sources/Swift/Impl/FlexLayout+Private.swift
index f2aabbae..af55be0c 100644
--- a/Sources/Swift/Impl/FlexLayout+Private.swift
+++ b/Sources/Swift/Impl/FlexLayout+Private.swift
@@ -7,7 +7,7 @@
 //
 
 import UIKit
-#if FLEXLAYOUT_SWIFT_PACKAGE
+#if FLEXLAYOUT_SWIFT_PACKAGE || SWIFT_PACKAGE
 import FlexLayoutYoga
 #endif
 
diff --git a/Sources/Swift/YGLayoutExtensions.swift b/Sources/Swift/YGLayoutExtensions.swift
index d51e3c12..ea3021a4 100644
--- a/Sources/Swift/YGLayoutExtensions.swift
+++ b/Sources/Swift/YGLayoutExtensions.swift
@@ -5,7 +5,7 @@
  * LICENSE file in the root directory of this source tree.
  */
 
-#if FLEXLAYOUT_SWIFT_PACKAGE
+#if FLEXLAYOUT_SWIFT_PACKAGE || SWIFT_PACKAGE
 import CoreGraphics
 import FlexLayoutYoga
 #endif
diff --git a/Sources/YogaKit/include/YogaKit/YGLayout+Private.h b/Sources/YogaKit/include/YogaKit/YGLayout+Private.h
index 07aad0c8..70cfe402 100644
--- a/Sources/YogaKit/include/YogaKit/YGLayout+Private.h
+++ b/Sources/YogaKit/include/YogaKit/YGLayout+Private.h
@@ -6,7 +6,7 @@
  */
 
 #import "YGLayout.h"
-#if FLEXLAYOUT_SWIFT_PACKAGE
+#if FLEXLAYOUT_SWIFT_PACKAGE || SWIFT_PACKAGE
 #import <yoga/Yoga.h>
 #else
 #import "Yoga.h"
diff --git a/Sources/YogaKit/include/YogaKit/YGLayout.h b/Sources/YogaKit/include/YogaKit/YGLayout.h
index 8faa3e82..a963fab1 100644
--- a/Sources/YogaKit/include/YogaKit/YGLayout.h
+++ b/Sources/YogaKit/include/YogaKit/YGLayout.h
@@ -7,7 +7,7 @@
 
 #import <UIKit/UIKit.h>
 
-#if FLEXLAYOUT_SWIFT_PACKAGE
+#if FLEXLAYOUT_SWIFT_PACKAGE || SWIFT_PACKAGE
 #import <yoga/YGEnums.h>
 #import <yoga/Yoga.h>
 #import <yoga/YGMacros.h>