diff --git a/src/com/android/settings/fuelgauge/batteryusage/BatteryDiffEntry.java b/src/com/android/settings/fuelgauge/batteryusage/BatteryDiffEntry.java
index 9e0c6d79196..c8577fc5772 100644
--- a/src/com/android/settings/fuelgauge/batteryusage/BatteryDiffEntry.java
+++ b/src/com/android/settings/fuelgauge/batteryusage/BatteryDiffEntry.java
@@ -272,8 +272,8 @@ public class BatteryDiffEntry {
 
     @VisibleForTesting
     void updateRestrictionFlagState() {
-        mValidForRestriction = true;
         if (!mBatteryHistEntry.isAppEntry()) {
+            mValidForRestriction = false;
             return;
         }
         final boolean isValidPackage =
diff --git a/tests/robotests/src/com/android/settings/fuelgauge/batteryusage/BatteryDiffEntryTest.java b/tests/robotests/src/com/android/settings/fuelgauge/batteryusage/BatteryDiffEntryTest.java
index 4883563f026..1085a6b6d73 100644
--- a/tests/robotests/src/com/android/settings/fuelgauge/batteryusage/BatteryDiffEntryTest.java
+++ b/tests/robotests/src/com/android/settings/fuelgauge/batteryusage/BatteryDiffEntryTest.java
@@ -168,8 +168,8 @@ public final class BatteryDiffEntryTest {
         assertThat(nameAndIcon.mName).isEqualTo(expectedName);
         assertThat(nameAndIcon.mIconId).isEqualTo(R.drawable.ic_settings_aod);
         // Verifies the restrictable flag in the cache.
-        assertThat(entry.mValidForRestriction).isTrue();
-        assertThat(BatteryDiffEntry.sValidForRestriction.get(entry.getKey())).isTrue();
+        assertThat(entry.mValidForRestriction).isFalse();
+        assertThat(BatteryDiffEntry.sValidForRestriction.get(entry.getKey())).isFalse();
     }
 
     @Test
@@ -194,8 +194,8 @@ public final class BatteryDiffEntryTest {
         assertThat(nameAndIcon.mName).isEqualTo(expectedName);
         assertThat(nameAndIcon.mIconId).isEqualTo(0);
         // Verifies the restrictable flag in the cache.
-        assertThat(entry.mValidForRestriction).isTrue();
-        assertThat(BatteryDiffEntry.sValidForRestriction.get(entry.getKey())).isTrue();
+        assertThat(entry.mValidForRestriction).isFalse();
+        assertThat(BatteryDiffEntry.sValidForRestriction.get(entry.getKey())).isFalse();
     }
 
     @Test