diff --git a/tests/Android.mk b/tests/Android.mk
index e264089..a70af48 100644
--- a/tests/Android.mk
+++ b/tests/Android.mk
@@ -23,7 +23,7 @@
     mockito-target-minus-junit4 \
     ub-uiautomator
 
-LOCAL_JAVA_LIBRARIES := android.test.runner.stubs
+LOCAL_JAVA_LIBRARIES := android.test.runner.stubs android.test.base.stubs
 
 LOCAL_AAPT_FLAGS := \
     --auto-add-overlay \