diff --git a/mParticle-UrbanAirship/MPKitUrbanAirship.h b/mParticle-UrbanAirship/MPKitUrbanAirship.h
index fa91a03..d16c30d 100755
--- a/mParticle-UrbanAirship/MPKitUrbanAirship.h
+++ b/mParticle-UrbanAirship/MPKitUrbanAirship.h
@@ -1,8 +1,13 @@
 #import <Foundation/Foundation.h>
 #if defined(__has_include) && __has_include(<mParticle_Apple_SDK/mParticle.h>)
-#import <mParticle_Apple_SDK/mParticle.h>
+    #import <mParticle_Apple_SDK/mParticle.h>
+    #import <mParticle_Apple_SDK/mParticle_Apple_SDK-Swift.h>
+#elif defined(__has_include) && __has_include(<mParticle_Apple_SDK_NoLocation/mParticle.h>)
+    #import <mParticle_Apple_SDK_NoLocation/mParticle.h>
+    #import <mParticle_Apple_SDK_NoLocation/mParticle_Apple_SDK-Swift.h>
 #else
-#import "mParticle.h"
+    #import "mParticle.h"
+    #import "mParticle_Apple_SDK-Swift.h"
 #endif
 
 @interface MPKitUrbanAirship : NSObject <MPKitProtocol>