diff --git a/apex/apex.go b/apex/apex.go
index 5b0def0..4924311 100644
--- a/apex/apex.go
+++ b/apex/apex.go
@@ -1418,7 +1418,7 @@
 var _ cc.Coverage = (*apexBundle)(nil)
 
 // Implements cc.Coverage
-func (a *apexBundle) IsNativeCoverageNeeded(ctx android.BaseModuleContext) bool {
+func (a *apexBundle) IsNativeCoverageNeeded(ctx android.IncomingTransitionContext) bool {
 	return ctx.DeviceConfig().NativeCoverageEnabled()
 }