diff --git a/rust/androidmk.go b/rust/androidmk.go
index e0cb3ce..17fd2d8 100644
--- a/rust/androidmk.go
+++ b/rust/androidmk.go
@@ -63,10 +63,8 @@
 				entries.AddStrings("LOCAL_SHARED_LIBRARIES", mod.transitiveAndroidMkSharedLibs.ToList()...)
 				entries.AddStrings("LOCAL_STATIC_LIBRARIES", mod.Properties.AndroidMkStaticLibs...)
 				entries.AddStrings("LOCAL_SOONG_LINK_TYPE", mod.makeLinkType)
-				if mod.InVendor() {
-					entries.SetBool("LOCAL_IN_VENDOR", true)
-				} else if mod.InProduct() {
-					entries.SetBool("LOCAL_IN_PRODUCT", true)
+				if mod.UseVndk() {
+					entries.SetBool("LOCAL_USE_VNDK", true)
 				}
 				android.SetAconfigFileMkEntries(mod.AndroidModuleBase(), entries, mod.mergedAconfigFiles)
 			},