diff --git a/ostd/src/cpu/mod.rs b/ostd/src/cpu/mod.rs
index 22ea2a012..bf90ce3dd 100644
--- a/ostd/src/cpu/mod.rs
+++ b/ostd/src/cpu/mod.rs
@@ -7,14 +7,10 @@ pub mod set;
 
 pub use set::{AtomicCpuSet, CpuSet};
 
-cfg_if::cfg_if! {
-    if #[cfg(target_arch = "x86_64")] {
-        pub use crate::arch::x86::cpu::*;
-    } else if #[cfg(target_arch = "riscv64")] {
-        pub use crate::arch::riscv::cpu::*;
-    }
-}
-
+#[cfg(target_arch = "riscv64")]
+pub use crate::arch::riscv::cpu::*;
+#[cfg(target_arch = "x86_64")]
+pub use crate::arch::x86::cpu::*;
 use crate::{cpu_local_cell, task::atomic_mode::InAtomicMode};
 
 /// The ID of a CPU in the system.