1 files changed, 4 insertions, 8 deletions
diff --git a/src/java/com/android/internal/telephony/metrics/DataStallRecoveryStats.java b/src/java/com/android/internal/telephony/metrics/DataStallRecoveryStats.java
index 0fd97ba25e..387495e109 100644
--- a/src/java/com/android/internal/telephony/metrics/DataStallRecoveryStats.java
+++ b/src/java/com/android/internal/telephony/metrics/DataStallRecoveryStats.java
@@ -42,7 +42,7 @@ import com.android.internal.telephony.subscription.SubscriptionInfoInternal;
 import com.android.internal.telephony.subscription.SubscriptionManagerService;
 import com.android.telephony.Rlog;
 
-import java.util.List;
+import java.util.Set;
 
 /**
  * Generates metrics related to data stall recovery events per phone ID for the pushed atom.
@@ -103,8 +103,9 @@ public class DataStallRecoveryStats {
         dataNetworkController.registerDataNetworkControllerCallback(
                 new DataNetworkControllerCallback(mHandler::post) {
                 @Override
-                public void onInternetDataNetworkConnected(
-                        @NonNull List<DataNetwork> internetNetworks) {
+                public void onConnectedInternetDataNetworksChanged(
+                        @NonNull Set<DataNetwork> internetNetworks) {
+                    mIfaceName = null;
                     for (DataNetwork dataNetwork : internetNetworks) {
                         mIfaceName = dataNetwork.getLinkProperties().getInterfaceName();
                         break;
@@ -112,11 +113,6 @@ public class DataStallRecoveryStats {
                 }
 
                 @Override
-                public void onInternetDataNetworkDisconnected() {
-                    mIfaceName = null;
-                }
-
-                @Override
                 public void onPhysicalLinkStatusChanged(@LinkStatus int status) {
                     mInternetLinkStatus = status;
                 }