diff --git a/gradle.properties b/gradle.properties
index 3fff6c9ba93ca369240e5d08a8f0cb2460277a43..d11d9555972860ca7434e98e7d843dd5d4a1ff67 100644
--- a/gradle.properties
+++ b/gradle.properties
@@ -1,6 +1,6 @@
 GROUP=org.briarproject
 POM_ARTIFACT_ID=dont-kill-me-lib
-VERSION_NAME=0.2.3
+VERSION_NAME=0.2.4
 
 POM_NAME=Do not kill me library
 POM_DESCRIPTION=An Android library helping to keep a foreground service with wake-locks running. No other use-cases considered.
diff --git a/lib/src/main/java/org/briarproject/android/dontkillmelib/DozeHelperImpl.kt b/lib/src/main/java/org/briarproject/android/dontkillmelib/DozeHelperImpl.kt
index 690eb7e08d5a20628c6817493cc76417bea16eab..b6937e1c858c384d90d69a450a56b7412a20d6c9 100644
--- a/lib/src/main/java/org/briarproject/android/dontkillmelib/DozeHelperImpl.kt
+++ b/lib/src/main/java/org/briarproject/android/dontkillmelib/DozeHelperImpl.kt
@@ -6,8 +6,9 @@ public class DozeHelperImpl : DozeHelper {
     override fun needToShowDoNotKillMeFragment(context: Context): Boolean {
         val appContext = context.applicationContext
         return DozeUtils.needsDozeWhitelisting(appContext) ||
-            HuaweiUtils.protectedAppsNeedsToBeShown(appContext) ||
-            HuaweiUtils.appLaunchNeedsToBeShown(appContext) ||
-            XiaomiUtils.isXiaomiOrRedmiDevice
+                HuaweiUtils.protectedAppsNeedsToBeShown(appContext) ||
+                HuaweiUtils.appLaunchNeedsToBeShown(appContext) ||
+                XiaomiUtils.xiaomiRecentAppsNeedsToBeShown ||
+                XiaomiUtils.xiaomiLockAppsNeedsToBeShown(appContext)
     }
 }