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 38e1f113c1..9499d98eac 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
@@ -107,12 +107,13 @@ public NewSegmentLayout(final Context context, final AttributeSet attributeSet,
      */
     private void initializeButton(final Context context, final String resourceIdentifierName,
                                   final ButtonOnClickHandlerFunction handler, final String debugMessage) {
-        final ImageButton button = findViewById(getResourceIdentifier(context, resourceIdentifierName, "id"));
+        ImageButton button = findViewById(getResourceIdentifier(context, resourceIdentifierName, "id"));
 
-        // 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();
@@ -121,7 +122,7 @@ private void initializeButton(final Context context, final String resourceIdenti
     }
 
     @FunctionalInterface
-    public interface ButtonOnClickHandlerFunction {
+    private interface ButtonOnClickHandlerFunction {
         void apply();
     }
-}
+}
\ No newline at end of file