diff --git a/ApiDocs.bp b/ApiDocs.bp
index c6a70d9..707b8c5 100644
--- a/ApiDocs.bp
+++ b/ApiDocs.bp
@@ -89,9 +89,9 @@
         ":updatable-media-srcs",
 
         // No longer part of the stubs, but are included in the docs.
-        "test-base/src/**/*.java",
-        "test-mock/src/**/*.java",
-        "test-runner/src/**/*.java",
+        ":android-test-base-sources",
+        ":android-test-mock-sources",
+        ":android-test-runner-sources",
     ],
     libs: framework_docs_only_libs,
     create_doc_stubs: true,