diff --git a/apex/apex.go b/apex/apex.go
index 29d59e5..3bc7d12 100644
--- a/apex/apex.go
+++ b/apex/apex.go
@@ -2386,7 +2386,7 @@
 				ProfilePathOnHost:             info.ProfilePathOnHost(),
 				LibraryNameToDexJarPathOnHost: info.DexBootJarPathMap(),
 			}
-			ctx.SetProvider(android.ApexExportsInfoProvider, exports)
+			android.SetProvider(ctx, android.ApexExportsInfoProvider, exports)
 		}
 	})
 }
diff --git a/apex/prebuilt.go b/apex/prebuilt.go
index 1f57b63..188875a 100644
--- a/apex/prebuilt.go
+++ b/apex/prebuilt.go
@@ -790,7 +790,7 @@
 			ProfilePathOnHost:             di.PrebuiltExportPath(java.ProfileInstallPathInApex),
 			LibraryNameToDexJarPathOnHost: javaModuleToDexPath,
 		}
-		ctx.SetProvider(android.ApexExportsInfoProvider, exports)
+		android.SetProvider(ctx, android.ApexExportsInfoProvider, exports)
 	} else {
 		ctx.ModuleErrorf(err.Error())
 	}