diff --git a/tests/unit/Android.bp b/tests/unit/Android.bp
index cc003ea..a38684c 100644
--- a/tests/unit/Android.bp
+++ b/tests/unit/Android.bp
@@ -32,6 +32,7 @@
         "truth-prebuilt",
         "androidx.test.uiautomator_uiautomator",
         "kotlinx_coroutines_test",
+        "flag-junit-base",
         // Don't add SettingsLib libraries here - you can use them directly as they are in the
         // instrumented Settings app.
     ],