diff --git a/tests/Android.bp b/tests/Android.bp
index f3a2e37..690a1b6 100644
--- a/tests/Android.bp
+++ b/tests/Android.bp
@@ -25,7 +25,7 @@
     name: "ThemePickerTests_srcs",
 
     visibility: [
-        "//visibility:public"
+        "//visibility:public",
     ],
 
     srcs: [
@@ -62,6 +62,7 @@
         "junit",
         "kotlinx_coroutines_test",
         "truth",
+        "flag-junit",
     ],
     libs: [
         "android.test.runner",