diff --git a/filesystem/android_device.go b/filesystem/android_device.go
index a616ee0..ebfe43c 100644
--- a/filesystem/android_device.go
+++ b/filesystem/android_device.go
@@ -927,7 +927,7 @@
 	apkCerts := []string{}
 	for _, installedModule := range allInstalledModules {
 		partition := ""
-		if commonInfo, ok := android.OtherModuleProvider(ctx, installedModule, android.CommonModuleInfoKey); ok {
+		if commonInfo, ok := android.OtherModuleProvider(ctx, installedModule, android.CommonModuleInfoProvider); ok {
 			partition = commonInfo.PartitionTag
 		} else {
 			ctx.ModuleErrorf("%s does not set CommonModuleInfoKey", installedModule.Name())