diff --git a/java/dex.go b/java/dex.go
index 8363a7d..d0ca06e 100644
--- a/java/dex.go
+++ b/java/dex.go
@@ -140,7 +140,7 @@
 }
 
 func (j *Module) compileDex(ctx android.ModuleContext, flags javaBuilderFlags,
-	classesJar android.Path, jarName string) android.Path {
+	classesJar android.Path, jarName string) android.ModuleOutPath {
 
 	useR8 := Bool(j.deviceProperties.Optimize.Enabled)