diff --git a/YTLitePlus.xm b/YTLitePlus.xm
index f372a56..a0f0c24 100644
--- a/YTLitePlus.xm
+++ b/YTLitePlus.xm
@@ -631,10 +631,7 @@ BOOL isTabSelected = NO;
 // Miscellaneous
 %group giPadLayout // https://github.com/LillieH001/YouTube-Reborn
 %hook UIDevice
-- (long long)userInterfaceIdiom {
-    return YES;
-}
-- (NSInteger)userInterfaceIdiom {
+- (UIUserInterfaceIdiom)userInterfaceIdiom {
     return UIUserInterfaceIdiomPad;
 }
 %end