1 files changed, 4 insertions, 4 deletions
diff --git a/Settings/src/com/android/tv/settings/system/development/OemUnlockPreferenceController.java b/Settings/src/com/android/tv/settings/system/development/OemUnlockPreferenceController.java
index 3e46ab9a7..e07bcf98c 100644
--- a/Settings/src/com/android/tv/settings/system/development/OemUnlockPreferenceController.java
+++ b/Settings/src/com/android/tv/settings/system/development/OemUnlockPreferenceController.java
@@ -24,7 +24,7 @@ import android.service.oemlock.OemLockManager;
 import android.text.TextUtils;
 import android.util.Log;
 
-import androidx.preference.SwitchPreference;
+import androidx.preference.TwoStatePreference;
 
 import com.android.tv.settings.R;
 
@@ -34,9 +34,9 @@ public class OemUnlockPreferenceController implements OemUnlockDialog.Callback {
 
     private final OemLockManager mOemLockManager;
     private final UserManager mUserManager;
-    private final SwitchPreference mPreference;
+    private final TwoStatePreference mPreference;
 
-    public OemUnlockPreferenceController(Context context, SwitchPreference preference) {
+    public OemUnlockPreferenceController(Context context, TwoStatePreference preference) {
         if (!TextUtils.equals(SystemProperties.get("ro.oem_unlock_supported", "0"), "1")) {
             mOemLockManager = null;
             Log.w(TAG, "oem_unlock not supported.");
@@ -48,7 +48,7 @@ public class OemUnlockPreferenceController implements OemUnlockDialog.Callback {
         updateState();
     }
 
-    public SwitchPreference getPreference() {
+    public TwoStatePreference getPreference() {
         return mPreference;
     }