diff --git a/tests/unittests/Android.bp b/tests/unittests/Android.bp
index 595b4be..6ac0586 100644
--- a/tests/unittests/Android.bp
+++ b/tests/unittests/Android.bp
@@ -12,14 +12,17 @@
         "mockito-target",
         "compatibility-device-util-axt",
         "androidx.test.rules",
-        "truth-prebuilt",
+        "truth",
     ],
     libs: [
         "android.test.runner",
         "android.test.base",
         "android.test.mock",
     ],
-    srcs: ["src/**/*.java", ":mms-service-srcs"],
+    srcs: [
+        "src/**/*.java",
+        ":mms-service-srcs",
+    ],
     platform_apis: true,
     test_suites: ["device-tests"],
     certificate: "platform",