diff --git a/tests/Android.mk b/tests/Android.mk
index b43533f..15de9b0 100644
--- a/tests/Android.mk
+++ b/tests/Android.mk
@@ -25,7 +25,7 @@
 
 LOCAL_MODULE_TAGS := tests
 
-LOCAL_JAVA_LIBRARIES := telephony-common android-support-test
+LOCAL_JAVA_LIBRARIES := telephony-common android-support-test android.test.base
 
 LOCAL_INSTRUMENTATION_FOR := TeleService
 
@@ -33,8 +33,7 @@
         android-support-test \
         mockito-target-minus-junit4 \
         espresso-core \
-        truth-prebuilt \
-        legacy-android-test
+        truth-prebuilt
 
 LOCAL_COMPATIBILITY_SUITE := device-tests