diff --git a/src/lib.rs b/src/lib.rs
index 2571f81a6cfbd99b5583d452a4e6149627702a8f..426b6849624893906d345790453eb40f3dfbf7b1 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -112,7 +112,7 @@ cfg_if! {
     } else if #[cfg(all(target_env = "sgx", target_vendor = "fortanix"))] {
         mod sgx;
         pub use sgx::*;
-    } else if #[cfg(target_env = "wasi")] {
+    } else if #[cfg(any(target_env = "wasi", target_os = "wasi"))] {
         mod wasi;
         pub use wasi::*;
     } else {