Update uYouPlus.xm

arichornlover · Nov 21, 2024 · 7b1a1a7 · 7b1a1a7
1 parent 3e922ea
commit 7b1a1a7
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/Sources/uYouPlus.xm b/Sources/uYouPlus.xm
@@ -608,9 +608,11 @@ static NSMutableArray <YTIItemSectionRenderer *> *filteredArray(NSArray <YTIItem
 %hook YTHeaderLogoController
 - (void)setTopbarLogoRenderer:(YTITopbarLogoRenderer *)renderer {
     // Modify the type of the icon before setting the renderer
-    YTIIcon *icon = [%c(YTIIcon) new];
-    icon.iconType = YT_PREMIUM_LOGO; // magic number (537) for Premium icon, hopefully it doesnt change. 158 (YT_DEFAULT_LOGO) is default logo.
-    renderer.iconImage = icon;
+    YTIIcon *icon = renderer.iconImage;
+    if (icon) {
+        icon.iconType = YT_PREMIUM_LOGO; // magic number (537) for Premium icon, hopefully it doesnt change. 158 (YT_DEFAULT_LOGO) is default logo.
+        }
+    // Use this modified renderer
     %orig;
 }
 // For when spoofing before 18.34.5