diff --git a/apex/apex.go b/apex/apex.go
index 69547c3..e01406b 100644
--- a/apex/apex.go
+++ b/apex/apex.go
@@ -3703,6 +3703,8 @@
 	commonAttrs := android.CommonAttributes{}
 	if a.testApex {
 		commonAttrs.Testonly = proptools.BoolPtr(true)
+		// Set the api_domain of the test apex
+		attrs.Base_apex_name = proptools.StringPtr(cc.GetApiDomain(a.Name()))
 	}
 
 	return attrs, props, commonAttrs