diff --git a/java/java.go b/java/java.go
index f934721..8750dbf 100644
--- a/java/java.go
+++ b/java/java.go
@@ -29,14 +29,6 @@
 	"android/soong/common"
 )
 
-type Config interface {
-	SrcDir() string
-	PrebuiltOS() string
-	HostBinTool(string) (string, error)
-	HostJavaTool(string) (string, error)
-	Getenv(string) string
-}
-
 // TODO:
 // Autogenerated files:
 //  AIDL
@@ -266,11 +258,11 @@
 			dxFlags = append(dxFlags, "--no-locals")
 		}
 
-		if ctx.Config().(Config).Getenv("NO_OPTIMIZE_DX") != "" {
+		if ctx.AConfig().Getenv("NO_OPTIMIZE_DX") != "" {
 			dxFlags = append(dxFlags, "--no-optimize")
 		}
 
-		if ctx.Config().(Config).Getenv("GENERATE_DEX_DEBUG") != "" {
+		if ctx.AConfig().Getenv("GENERATE_DEX_DEBUG") != "" {
 			dxFlags = append(dxFlags,
 				"--debug",
 				"--verbose",