diff --git a/tokio/src/task/join_set.rs b/tokio/src/task/join_set.rs
index 24d3717a26b..66e53a129d1 100644
--- a/tokio/src/task/join_set.rs
+++ b/tokio/src/task/join_set.rs
@@ -10,7 +10,7 @@ use std::pin::Pin;
 use std::task::{Context, Poll};
 
 use crate::runtime::Handle;
-#[cfg(unstable)]
+#[cfg(tokio_unstable)]
 use crate::task::Id;
 use crate::task::{AbortHandle, JoinError, JoinHandle, LocalSet};
 use crate::util::IdleNotifiedSet;
@@ -341,7 +341,7 @@ impl<T: 'static> JoinSet<T> {
     ///
     /// [coop budget]: crate::task#cooperative-scheduling
     /// [task ID]: crate::task::Id
-    #[cfg(unstable)]
+    #[cfg(tokio_unstable)]
     fn poll_join_next_with_id(
         &mut self,
         cx: &mut Context<'_>,