diff --git a/CYRKeyboardButton/CYRKeyboardButtonView.m b/CYRKeyboardButton/CYRKeyboardButtonView.m
index cb0ba2a..b7ec756 100644
--- a/CYRKeyboardButton/CYRKeyboardButtonView.m
+++ b/CYRKeyboardButton/CYRKeyboardButtonView.m
@@ -53,7 +53,7 @@ - (instancetype)initWithKeyboardButton:(CYRKeyboardButton *)button type:(CYRKeyb
 {
     CGRect frame = [UIScreen mainScreen].bounds;
     
-    if (UIInterfaceOrientationIsLandscape([UIApplication sharedApplication].statusBarOrientation)) {
+    if ([UIScreen mainScreen].bounds.size.width > [UIScreen mainScreen].bounds.size.height) {
         frame = CGRectMake(0, 0, CGRectGetHeight(frame), CGRectGetWidth(frame));
     }