diff --git a/sdk/testing.go b/sdk/testing.go
index 77ba2e6..3dc88c1 100644
--- a/sdk/testing.go
+++ b/sdk/testing.go
@@ -78,10 +78,8 @@
 	ctx.RegisterModuleType("package", android.PackageFactory)
 
 	// from java package
-	ctx.RegisterModuleType("android_app_certificate", java.AndroidAppCertificateFactory)
-	ctx.RegisterModuleType("java_defaults", java.DefaultsFactory)
-	ctx.RegisterModuleType("java_library", java.LibraryFactory)
-	ctx.RegisterModuleType("java_import", java.ImportFactory)
+	java.RegisterJavaBuildComponents(ctx)
+	java.RegisterAppBuildComponents(ctx)
 	ctx.RegisterModuleType("droidstubs", java.DroidstubsFactory)
 	ctx.RegisterModuleType("prebuilt_stubs_sources", java.PrebuiltStubsSourcesFactory)