diff --git a/src/com/android/settings/network/telephony/DataUsagePreferenceController.kt b/src/com/android/settings/network/telephony/DataUsagePreferenceController.kt
index 14adf6f1918..88da325c3e6 100644
--- a/src/com/android/settings/network/telephony/DataUsagePreferenceController.kt
+++ b/src/com/android/settings/network/telephony/DataUsagePreferenceController.kt
@@ -30,6 +30,7 @@ import androidx.preference.Preference
 import androidx.preference.PreferenceScreen
 import com.android.settings.R
 import com.android.settings.datausage.DataUsageUtils
+import com.android.settings.datausage.lib.DataUsageLib
 import com.android.settingslib.net.DataUsageController
 import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.launch
@@ -96,7 +97,7 @@ class DataUsagePreferenceController(context: Context, key: String) :
 
     private fun getNetworkTemplate(): NetworkTemplate? = when {
         SubscriptionManager.isValidSubscriptionId(mSubId) -> {
-            DataUsageUtils.getMobileTemplate(mContext, mSubId)
+            DataUsageLib.getMobileTemplate(mContext, mSubId)
         }
 
         else -> null
diff --git a/tests/spa_unit/src/com/android/settings/network/telephony/DataUsagePreferenceControllerTest.kt b/tests/spa_unit/src/com/android/settings/network/telephony/DataUsagePreferenceControllerTest.kt
index a9d343fa0b5..a42e41ed77e 100644
--- a/tests/spa_unit/src/com/android/settings/network/telephony/DataUsagePreferenceControllerTest.kt
+++ b/tests/spa_unit/src/com/android/settings/network/telephony/DataUsagePreferenceControllerTest.kt
@@ -32,6 +32,7 @@ import com.android.dx.mockito.inline.extended.ExtendedMockito
 import com.android.settings.core.BasePreferenceController.AVAILABLE
 import com.android.settings.core.BasePreferenceController.AVAILABLE_UNSEARCHABLE
 import com.android.settings.datausage.DataUsageUtils
+import com.android.settings.datausage.lib.DataUsageLib
 import com.android.settingslib.net.DataUsageController
 import com.android.settingslib.net.DataUsageController.DataUsageInfo
 import com.android.settingslib.spa.testutils.waitUntil
@@ -80,13 +81,14 @@ class DataUsagePreferenceControllerTest {
             .initMocks(this)
             .mockStatic(SubscriptionManager::class.java)
             .spyStatic(DataUsageUtils::class.java)
+            .spyStatic(DataUsageLib::class.java)
             .strictness(Strictness.LENIENT)
             .startMocking()
 
         whenever(SubscriptionManager.isValidSubscriptionId(SUB_ID)).thenReturn(true)
         ExtendedMockito.doReturn(true).`when` { DataUsageUtils.hasMobileData(context) }
         ExtendedMockito.doReturn(networkTemplate)
-            .`when` { DataUsageUtils.getMobileTemplate(context, SUB_ID) }
+            .`when` { DataUsageLib.getMobileTemplate(context, SUB_ID) }
         preference.key = TEST_KEY
         whenever(preferenceScreen.findPreference<Preference>(TEST_KEY)).thenReturn(preference)