Merge branch 'master' of github.com:AzureAD/azure-activedirectory-lib…

…rary-for-objc
AzureAD · Sep 30, 2014 · 37cda1a · 37cda1a
2 parents 7b78fc3 + ee86759
commit 37cda1a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ADALiOS/ADALiOS/ADAuthenticationBroker.m b/ADALiOS/ADALiOS/ADAuthenticationBroker.m
@@ -367,7 +367,7 @@ - (void)webAuthenticationDidCompleteWithURL:(NSURL *)endURL
     if ( nil != _authenticationViewController)
     {
         // Dismiss the authentication view and dispatch the completion block
-        [[UIApplication adCurrentViewController] dismissViewControllerAnimated:YES completion:^{
+        [parentController dismissViewControllerAnimated:YES completion:^{
             [self dispatchCompletionBlock:nil URL:endURL];
         }];
     }
@@ -390,7 +390,7 @@ - (void)webAuthenticationDidFailWithError:(NSError *)error
     if ( nil != _authenticationViewController)
     {
         // Dismiss the authentication view and dispatch the completion block
-        [[UIApplication adCurrentViewController] dismissViewControllerAnimated:YES completion:^{
+        [parentController dismissViewControllerAnimated:YES completion:^{
             [self dispatchCompletionBlock:adError URL:nil];
         }];
     }