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