diff --git a/rust-toolchain b/rust-toolchain
index d0a4304f5721..55b557c0ada9 100644
--- a/rust-toolchain
+++ b/rust-toolchain
@@ -1,4 +1,4 @@
 [toolchain]
-channel = "nightly-2025-04-30"
+channel = "nightly-2025-05-01"
 components = ["rust-src", "rustc-dev", "llvm-tools"]
 profile = "minimal"
diff --git a/src/driver/jit.rs b/src/driver/jit.rs
index 403e5403ccc6..b1f185b551c3 100644
--- a/src/driver/jit.rs
+++ b/src/driver/jit.rs
@@ -6,6 +6,7 @@ use std::os::raw::{c_char, c_int};
 
 use cranelift_jit::{JITBuilder, JITModule};
 use rustc_codegen_ssa::CrateInfo;
+use rustc_middle::middle::codegen_fn_attrs::CodegenFnAttrFlags;
 use rustc_middle::mir::mono::MonoItem;
 use rustc_session::Session;
 use rustc_span::sym;