diff --git a/apex/apex.go b/apex/apex.go
index 04ed269..8e9bd04 100644
--- a/apex/apex.go
+++ b/apex/apex.go
@@ -865,9 +865,7 @@
 		return
 	}
 
-	cur := mctx.Module().(interface {
-		DepIsInSameApex(android.BaseModuleContext, android.Module) bool
-	})
+	cur := mctx.Module().(android.DepIsInSameApex)
 
 	mctx.VisitDirectDeps(func(child android.Module) {
 		depName := mctx.OtherModuleName(child)