diff --git a/Examples/CodePushDemoApp/iOS/CodePushDemoAppTests/CodePushDemoAppTests.m b/Examples/CodePushDemoApp/iOS/CodePushDemoAppTests/CodePushDemoAppTests.m
index 457dc5c..5dc746f 100644
--- a/Examples/CodePushDemoApp/iOS/CodePushDemoAppTests/CodePushDemoAppTests.m
+++ b/Examples/CodePushDemoApp/iOS/CodePushDemoAppTests/CodePushDemoAppTests.m
@@ -12,6 +12,7 @@
 
 #import <React/RCTLog.h>
 #import <React/RCTRootView.h>
+#import <React/RCTUtils.h>
 
 #define TIMEOUT_SECONDS 600
 #define TEXT_TO_LOOK_FOR @"Welcome to React Native!"
@@ -37,7 +38,7 @@
 
 - (void)testRendersWelcomeScreen
 {
-  UIViewController *vc = [[[[UIApplication sharedApplication] delegate] window] rootViewController];
+  UIViewController *vc = [[[RCTSharedApplication() delegate] window] rootViewController];
   NSDate *date = [NSDate dateWithTimeIntervalSinceNow:TIMEOUT_SECONDS];
   BOOL foundElement = NO;