diff --git a/DeviceDoze/src/com/advanced/settings/doze/DozeSettingsFragment.java b/DeviceDoze/src/com/advanced/settings/doze/DozeSettingsFragment.java
index d99f2b2d..96f268ac 100644
--- a/DeviceDoze/src/com/advanced/settings/doze/DozeSettingsFragment.java
+++ b/DeviceDoze/src/com/advanced/settings/doze/DozeSettingsFragment.java
@@ -30,13 +30,13 @@ import androidx.preference.SwitchPreference;
 import androidx.preference.Preference;
 import androidx.preference.PreferenceCategory;
 import androidx.preference.Preference.OnPreferenceChangeListener;
-import android.widget.Switch;
+import android.widget.CompoundButton;
+import android.widget.CompoundButton.OnCheckedChangeListener;
 
 import com.android.settingslib.widget.MainSwitchPreference;
-import com.android.settingslib.widget.OnMainSwitchChangeListener;
 
 public class DozeSettingsFragment extends PreferenceFragment implements OnPreferenceChangeListener,
-        OnMainSwitchChangeListener {
+        OnCheckedChangeListener {
 
     private MainSwitchPreference mSwitchBar;
 
@@ -106,7 +106,7 @@ public class DozeSettingsFragment extends PreferenceFragment implements OnPrefer
     }
 
     @Override
-    public void onSwitchChanged(Switch switchView, boolean isChecked) {
+    public void onCheckedChanged(CompoundButton buttonView, boolean isChecked) {
         Utils.enableDoze(getActivity(), isChecked);
         Utils.checkDozeService(getActivity());