diff --git a/src/ios/YoikScreenOrientation.h b/src/ios/YoikScreenOrientation.h
index 95da6e8..f226a93 100644
--- a/src/ios/YoikScreenOrientation.h
+++ b/src/ios/YoikScreenOrientation.h
@@ -23,7 +23,6 @@ SOFTWARE.
  */
 
 #import <Cordova/CDVPlugin.h>
-#import <Cordova/CDVShared.h>
 
 @interface YoikScreenOrientation : CDVPlugin
 
diff --git a/src/ios/YoikScreenOrientation.m b/src/ios/YoikScreenOrientation.m
index f2dd444..51e90f0 100644
--- a/src/ios/YoikScreenOrientation.m
+++ b/src/ios/YoikScreenOrientation.m
@@ -95,7 +95,7 @@ SOFTWARE.
 
 @implementation ForcedViewController
 
-- (NSUInteger) supportedInterfaceOrientations
+- (UIInterfaceOrientationMask) supportedInterfaceOrientations
 {
     if ([self.calledWith rangeOfString:@"portrait"].location != NSNotFound) {
         return UIInterfaceOrientationMaskPortrait;