diff --git a/Android.mk b/Android.mk
index 5884c5c..577052d 100644
--- a/Android.mk
+++ b/Android.mk
@@ -62,6 +62,8 @@
 
 
 
+LOCAL_JAVA_LIBRARIES := junit
+
 LOCAL_JAVACFLAGS := -Xlint:deprecation -Xlint:unchecked
 
 LOCAL_AAPT_FLAGS := --auto-add-overlay \
diff --git a/tests/func/Android.mk b/tests/func/Android.mk
index e89ba25..b0c5ad1 100644
--- a/tests/func/Android.mk
+++ b/tests/func/Android.mk
@@ -14,6 +14,8 @@
     tv-test-common \
     ub-uiautomator \
 
+LOCAL_JAVA_LIBRARIES := legacy-android-test
+
 LOCAL_INSTRUMENTATION_FOR := LiveTv
 
 LOCAL_SDK_VERSION := current
diff --git a/tests/jank/Android.mk b/tests/jank/Android.mk
index b71ea1b..210a75e 100644
--- a/tests/jank/Android.mk
+++ b/tests/jank/Android.mk
@@ -15,6 +15,8 @@
     ub-janktesthelper \
     ub-uiautomator \
 
+LOCAL_JAVA_LIBRARIES := legacy-android-test
+
 LOCAL_INSTRUMENTATION_FOR := LiveTv
 
 LOCAL_SDK_VERSION := current
diff --git a/tests/unit/Android.mk b/tests/unit/Android.mk
index 7ed100d..f53220d 100644
--- a/tests/unit/Android.mk
+++ b/tests/unit/Android.mk
@@ -12,6 +12,8 @@
     mockito-target \
     tv-test-common \
 
+LOCAL_JAVA_LIBRARIES := android.test.mock.sdk legacy-android-test
+
 LOCAL_PACKAGE_NAME := TVUnitTests
 
 LOCAL_INSTRUMENTATION_FOR := LiveTv