diff --git a/src/com/android/settings/network/InternetPreferenceController.java b/src/com/android/settings/network/InternetPreferenceController.java
index 4ae4d279f43..0e95e2dcd8b 100644
--- a/src/com/android/settings/network/InternetPreferenceController.java
+++ b/src/com/android/settings/network/InternetPreferenceController.java
@@ -62,7 +62,8 @@ public class InternetPreferenceController extends AbstractPreferenceController i
     private InternetUpdater mInternetUpdater;
     private @InternetUpdater.InternetType int mInternetType;
 
-    private static Map<Integer, Integer> sIconMap = new HashMap<>();
+    @VisibleForTesting
+    static Map<Integer, Integer> sIconMap = new HashMap<>();
     static {
         sIconMap.put(INTERNET_APM, R.drawable.ic_airplanemode_active);
         sIconMap.put(INTERNET_APM_NETWORKS, R.drawable.ic_airplane_safe_networks_24dp);
diff --git a/tests/unit/src/com/android/settings/network/InternetPreferenceControllerTest.java b/tests/unit/src/com/android/settings/network/InternetPreferenceControllerTest.java
index eb97b0595c3..be8bf27f0e7 100644
--- a/tests/unit/src/com/android/settings/network/InternetPreferenceControllerTest.java
+++ b/tests/unit/src/com/android/settings/network/InternetPreferenceControllerTest.java
@@ -79,6 +79,7 @@ public class InternetPreferenceControllerTest {
         when(wifiManager.getWifiState()).thenReturn(WifiManager.WIFI_STATE_DISABLED);
 
         mController = new InternetPreferenceController(mContext, mock(Lifecycle.class));
+        mController.sIconMap.put(INTERNET_WIFI, 0);
         if (Looper.myLooper() == null) {
             Looper.prepare();
         }