1 files changed, 12 insertions, 17 deletions
diff --git a/tokio/src/sync/tests/mod.rs b/tokio/src/sync/tests/mod.rs
index 8da739f9..06d18e9a 100644
--- a/tokio/src/sync/tests/mod.rs
+++ b/tokio/src/sync/tests/mod.rs
@@ -1,17 +1,12 @@
-#[cfg(not(loom))]
-mod atomic_waker;
-
-#[cfg(loom)]
-mod loom_atomic_waker;
-
-#[cfg(loom)]
-mod loom_list;
-
-#[cfg(loom)]
-mod loom_mpsc;
-
-#[cfg(loom)]
-mod loom_oneshot;
-
-#[cfg(loom)]
-mod loom_semaphore;
+cfg_not_loom! {
+    mod atomic_waker;
+    mod semaphore;
+}
+
+cfg_loom! {
+    mod loom_atomic_waker;
+    mod loom_list;
+    mod loom_mpsc;
+    mod loom_oneshot;
+    mod loom_semaphore;
+}