diff --git a/src/com/android/settings/accessibility/AccessibilitySetupWizardUtils.java b/src/com/android/settings/accessibility/AccessibilitySetupWizardUtils.java
index 95b689b73eb..c5302143182 100644
--- a/src/com/android/settings/accessibility/AccessibilitySetupWizardUtils.java
+++ b/src/com/android/settings/accessibility/AccessibilitySetupWizardUtils.java
@@ -47,7 +47,7 @@ class AccessibilitySetupWizardUtils {
         layout.setIcon(icon);
         layout.setDividerInsets(Integer.MAX_VALUE, 0);
 
-        if (ThemeHelper.shouldApplyExtendedPartnerConfig(context)) {
+        if (ThemeHelper.shouldApplyMaterialYouStyle(context)) {
             final LinearLayout headerLayout = layout.findManagedViewById(R.id.sud_layout_header);
             if (headerLayout != null) {
                 headerLayout.setPadding(0, layout.getPaddingTop(), 0,