1 files changed, 4 insertions, 9 deletions
diff --git a/src/com/android/settings/security/LockdownButtonPreferenceController.java b/src/com/android/settings/security/LockdownButtonPreferenceController.java
index d0a49d7bf6..5b29868f06 100644
--- a/src/com/android/settings/security/LockdownButtonPreferenceController.java
+++ b/src/com/android/settings/security/LockdownButtonPreferenceController.java
@@ -19,30 +19,25 @@ package com.android.settings.security;
 import android.content.Context;
 import android.os.UserHandle;
 import android.provider.Settings;
-import androidx.preference.Preference;
-import androidx.preference.TwoStatePreference;
 
 import com.android.internal.widget.LockPatternUtils;
-import com.android.settings.core.BasePreferenceController;
 import com.android.settings.core.TogglePreferenceController;
 
 public class LockdownButtonPreferenceController extends TogglePreferenceController {
 
-    private static final String KEY_LOCKDOWN_ENALBED = "security_setting_lockdown_enabled";
-
     private final LockPatternUtils mLockPatternUtils;
 
-    public LockdownButtonPreferenceController(Context context) {
-        super(context, KEY_LOCKDOWN_ENALBED);
+    public LockdownButtonPreferenceController(Context context, String key) {
+        super(context, key);
         mLockPatternUtils = new LockPatternUtils(context);
     }
 
     @Override
     public int getAvailabilityStatus() {
         if (mLockPatternUtils.isSecure(UserHandle.myUserId())) {
-            return BasePreferenceController.AVAILABLE;
+            return AVAILABLE;
         } else {
-            return BasePreferenceController.DISABLED_FOR_USER;
+            return DISABLED_FOR_USER;
         }
     }