From 3d245dc784ff31566da176941743d6f82a267543 Mon Sep 17 00:00:00 2001
From: arichorn <78001398+arichorn@users.noreply.github.com>
Date: Sun, 1 Oct 2023 20:20:58 -0500
Subject: [PATCH] Improved some Options

---
 uYouPlus.xm | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/uYouPlus.xm b/uYouPlus.xm
index e3a2de2..127d091 100644
--- a/uYouPlus.xm
+++ b/uYouPlus.xm
@@ -745,16 +745,18 @@ static void replaceTab(YTIGuideResponse *response) {
         self.hidden = YES;
         self.opaque = YES;
         self.userInteractionEnabled = NO;
+        [self sizeToFit];
         [self setNeedsLayout];
-        [self layoutIfNeeded];
+        [self removeFromSuperview];
     }
 
 // Hide the Download Button under the Video Player - @arichorn
     if ((IsEnabled(@"hideAddToOfflineButton_enabled")) && ([self.accessibilityIdentifier isEqualToString:@"id.ui.add_to_offline.button"])) {
         self.hidden = YES;
         self.userInteractionEnabled = NO;
+        [self sizeToFit];
         [self setNeedsLayout];
-        [self layoutIfNeeded];
+        [self removeFromSuperview];
     }
 
 // Hide the Comment Section under the Video Player - @arichorn
@@ -762,8 +764,9 @@ static void replaceTab(YTIGuideResponse *response) {
         self.hidden = YES;
         self.opaque = YES;
         self.userInteractionEnabled = NO;
+        [self sizeToFit];
         [self setNeedsLayout];
-        [self layoutIfNeeded];
+        [self removeFromSuperview];
     }
 }
 %end