diff --git a/tests/cts/hostside/app/src/com/android/cts/net/hostside/DataSaverModeTest.java b/tests/cts/hostside/app/src/com/android/cts/net/hostside/DataSaverModeTest.java
index 8efea01..7ca302f 100644
--- a/tests/cts/hostside/app/src/com/android/cts/net/hostside/DataSaverModeTest.java
+++ b/tests/cts/hostside/app/src/com/android/cts/net/hostside/DataSaverModeTest.java
@@ -133,6 +133,7 @@
         assertDataSaverStatusOnBackground(RESTRICT_BACKGROUND_STATUS_ENABLED);
 
         assertsForegroundAlwaysHasNetworkAccess();
+        assertRestrictBackgroundChangedReceived(1);
         assertDataSaverStatusOnBackground(RESTRICT_BACKGROUND_STATUS_ENABLED);
 
         // UID policies live by the Highlander rule: "There can be only one".
@@ -147,16 +148,17 @@
         // Check status after removing blacklist.
         // ...re-enables first
         addRestrictBackgroundBlacklist(mUid);
+        assertRestrictBackgroundChangedReceived(4);
         assertDataSaverStatusOnBackground(RESTRICT_BACKGROUND_STATUS_ENABLED);
         assertsForegroundAlwaysHasNetworkAccess();
         // ... remove blacklist - access's still rejected because Data Saver is on
         removeRestrictBackgroundBlacklist(mUid);
-        assertRestrictBackgroundChangedReceived(3);
+        assertRestrictBackgroundChangedReceived(4);
         assertDataSaverStatusOnBackground(RESTRICT_BACKGROUND_STATUS_ENABLED);
         assertsForegroundAlwaysHasNetworkAccess();
         // ... finally, disable Data Saver
         setRestrictBackground(false);
-        assertRestrictBackgroundChangedReceived(4);
+        assertRestrictBackgroundChangedReceived(5);
         assertDataSaverStatusOnBackground(RESTRICT_BACKGROUND_STATUS_DISABLED);
         assertsForegroundAlwaysHasNetworkAccess();
     }