diff --git a/apex/apex.go b/apex/apex.go
index 67c0a19..f934134 100644
--- a/apex/apex.go
+++ b/apex/apex.go
@@ -2885,7 +2885,7 @@
 	}
 
 	var appEmbeddedJNILibs android.Paths
-	ctx.VisitDirectDeps(func(dep android.Module) {
+	ctx.VisitDirectDepsProxy(func(dep android.ModuleProxy) {
 		tag := ctx.OtherModuleDependencyTag(dep)
 		if !checkApexTag(tag) {
 			return