diff --git a/src/vmm/build.rs b/src/vmm/build.rs
index 3898b61c7..dbaac85af 100644
--- a/src/vmm/build.rs
+++ b/src/vmm/build.rs
@@ -1,6 +1,5 @@
 fn main() {
-    #[cfg(target_arch = "aarch64")]
-    {
+    if std::env::var("CARGO_CFG_TARGET_ARCH").as_deref() == Ok("aarch64") {
         let edk2_binary_path = std::env::var("KRUN_EDK2_BINARY_PATH").unwrap_or_else(|_| {
             format!(
                 "{}/../../edk2/KRUN_EFI.silent.fd",