diff --git a/tests/integration/Android.bp b/tests/integration/Android.bp
index ec16467..89b3968 100644
--- a/tests/integration/Android.bp
+++ b/tests/integration/Android.bp
@@ -24,7 +24,7 @@
         "androidx.test.rules",
         "mockito-target-extended-minus-junit4",
         "net-tests-utils",
-        "NetworkStackBase",
+        "NetworkStackApiCurrentLib",
         "testables",
     ],
     libs: [
diff --git a/tests/unit/Android.bp b/tests/unit/Android.bp
index 9c1c791..3081666 100644
--- a/tests/unit/Android.bp
+++ b/tests/unit/Android.bp
@@ -24,7 +24,7 @@
         "androidx.test.rules",
         "mockito-target-extended-minus-junit4",
         "net-tests-utils",
-        "NetworkStackBase",
+        "NetworkStackApiCurrentLib",
         "testables",
     ],
     libs: [