diff --git a/src/com/android/settings/security/applock/AppLockTogglePreferenceController.kt b/src/com/android/settings/security/applock/AppLockTogglePreferenceController.kt
index 3564110e43fc25228a23a2089d890b6995c0107b..e73fead8cf39a0e43c2361f50a0e25121bc03836 100644
--- a/src/com/android/settings/security/applock/AppLockTogglePreferenceController.kt
+++ b/src/com/android/settings/security/applock/AppLockTogglePreferenceController.kt
@@ -1,5 +1,6 @@
 /*
  * Copyright (C) 2022 FlamingoOS Project
+ * Copyright (C) 2024 The LibreMobileOS Foundation
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -17,7 +18,8 @@
 package com.android.settings.security.applock;
 
 import android.content.Context
-import android.widget.Switch
+import android.widget.CompoundButton;
+import android.widget.CompoundButton.OnCheckedChangeListener;
 
 import androidx.preference.Preference
 import androidx.preference.PreferenceScreen
@@ -25,13 +27,12 @@ import androidx.preference.PreferenceScreen
 import com.android.settings.R
 import com.android.settings.core.TogglePreferenceController
 import com.android.settingslib.widget.MainSwitchPreference
-import com.android.settingslib.widget.OnMainSwitchChangeListener
 
 abstract class AppLockTogglePreferenceController(
     context: Context,
     key: String,
 ) : TogglePreferenceController(context, key),
-    OnMainSwitchChangeListener {
+    OnCheckedChangeListener {
 
     override fun displayPreference(screen: PreferenceScreen) {
         super.displayPreference(screen)
@@ -41,7 +42,7 @@ abstract class AppLockTogglePreferenceController(
         }
     }
 
-    override fun onSwitchChanged(switchView: Switch, isChecked: Boolean) {
+    override fun onCheckedChanged(buttonView: CompoundButton, isChecked: Boolean) {
         setChecked(isChecked)
     }