diff --git a/build.rs b/build.rs
index 260e7ac..3d65ccc 100644
--- a/build.rs
+++ b/build.rs
@@ -24,10 +24,6 @@ fn main() {
         setup_compiler_flags(&mut build);
         build.compile("libc");
     }
-    if target_arch != "riscv64" && cfg!(feature = "dummy-atomic") {
-        println!("cargo:warning=This build script intentionally failed: feature `dummy-atomic` can't be used in non risc-v target");
-        std::process::exit(1);
-    }
 }
 
 fn setup_compiler_flags(build: &mut cc::Build) {
diff --git a/src/lib.rs b/src/lib.rs
index bf87577..64751ad 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -35,7 +35,7 @@ pub mod dynamic_loading;
 pub mod dynamic_loading_c_impl;
 #[cfg(feature = "allocator")]
 pub use buddy_alloc;
-#[cfg(feature = "dummy-atomic")]
+#[cfg(all(target_arch = "riscv64", feature = "dummy-atomic"))]
 pub mod dummy_atomic;
 #[cfg(feature = "log")]
 pub mod logger;