diff --git a/extensions/shared/src/main/java/app/revanced/extension/youtube/sponsorblock/ui/NewSegmentLayout.java b/extensions/shared/src/main/java/app/revanced/extension/youtube/sponsorblock/ui/NewSegmentLayout.java
index e0b6ae69d..829c79e4c 100644
--- a/extensions/shared/src/main/java/app/revanced/extension/youtube/sponsorblock/ui/NewSegmentLayout.java
+++ b/extensions/shared/src/main/java/app/revanced/extension/youtube/sponsorblock/ui/NewSegmentLayout.java
@@ -101,12 +101,13 @@ public final class NewSegmentLayout extends FrameLayout {
      */
     private void initializeButton(final Context context, final String resourceIdentifierName,
                                   final ButtonOnClickHandlerFunction handler, final String debugMessage) {
-        final ImageButton button = findViewById(getIdentifier(resourceIdentifierName, ResourceUtils.ResourceType.ID, context));
+        ImageButton button = findViewById(getIdentifier(resourceIdentifierName, ResourceUtils.ResourceType.ID, context));
 
-        // Add ripple effect
         button.setBackgroundResource(rippleEffectId);
-        RippleDrawable rippleDrawable = (RippleDrawable) button.getBackground();
-        rippleDrawable.setColor(rippleColorStateList);
+        RippleDrawable rippleDrawable = new RippleDrawable(
+                rippleColorStateList, null, null
+        );
+        button.setBackground(rippleDrawable);
 
         button.setOnClickListener((v) -> {
             handler.apply();
@@ -115,7 +116,7 @@ public final class NewSegmentLayout extends FrameLayout {
     }
 
     @FunctionalInterface
-    public interface ButtonOnClickHandlerFunction {
+    private interface ButtonOnClickHandlerFunction {
         void apply();
     }
 }