diff --git a/src/com/android/settings/display/AdaptiveSleepPreferenceController.java b/src/com/android/settings/display/AdaptiveSleepPreferenceController.java index dfe6826195e..c0c5e63bfa4 100644 --- a/src/com/android/settings/display/AdaptiveSleepPreferenceController.java +++ b/src/com/android/settings/display/AdaptiveSleepPreferenceController.java @@ -75,6 +75,14 @@ public class AdaptiveSleepPreferenceController { * Adds the controlled preference to the provided preference screen. */ public void addToScreen(PreferenceScreen screen) { + updatePreference(); + screen.addPreference(mPreference); + } + + /** + * Updates the appearance of the preference. + */ + public void updatePreference() { final EnforcedAdmin enforcedAdmin = mRestrictionUtils.checkIfRestrictionEnforced(mContext, UserManager.DISALLOW_CONFIG_SCREEN_TIMEOUT); if (enforcedAdmin != null) { @@ -82,7 +90,6 @@ public class AdaptiveSleepPreferenceController { } else { mPreference.setEnabled(hasSufficientPermission(mPackageManager)); } - screen.addPreference(mPreference); } @VisibleForTesting diff --git a/src/com/android/settings/display/ScreenTimeoutSettings.java b/src/com/android/settings/display/ScreenTimeoutSettings.java index 35c2cc0cfe2..676acf9671f 100644 --- a/src/com/android/settings/display/ScreenTimeoutSettings.java +++ b/src/com/android/settings/display/ScreenTimeoutSettings.java @@ -104,6 +104,7 @@ public class ScreenTimeoutSettings extends RadioButtonPickerFragment implements public void onStart() { super.onStart(); mAdaptiveSleepPermissionController.updateVisibility(); + mAdaptiveSleepController.updatePreference(); } @Override