diff --git a/crates/wasi/src/lib.rs b/crates/wasi/src/lib.rs
index 2eefcfccd2..f0ac46811c 100644
--- a/crates/wasi/src/lib.rs
+++ b/crates/wasi/src/lib.rs
@@ -99,7 +99,7 @@ necessary. Additionally [`Wasi::get_export`] can be used to do name-based
 resolution.",
                 },
             },
-            $async_mode: { wasi_snapshot_preview1::{ poll_oneoff, sched_yield } }
+            $async_mode: *
         });
     }
     pub mod preview_0 {
@@ -127,7 +127,7 @@ necessary. Additionally [`Wasi::get_export`] can be used to do name-based
 resolution.",
                 },
             },
-            $async_mode: { wasi_unstable::{ poll_oneoff, sched_yield } }
+            $async_mode: *
         });
     }
 }