diff --git a/library/contracts/safety/Cargo.toml b/library/contracts/safety/Cargo.toml
index e51487e7266e9..5e912e117224d 100644
--- a/library/contracts/safety/Cargo.toml
+++ b/library/contracts/safety/Cargo.toml
@@ -15,3 +15,7 @@ proc-macro2 = "1.0"
 proc-macro-error = "1.0.4"
 quote = "1.0.20"
 syn = { version = "2.0.18", features = ["full"] }
+
+
+[lints.rust]
+unexpected_cfgs = { level = "warn", check-cfg = ['cfg(kani_host)'] }
\ No newline at end of file
diff --git a/library/contracts/safety/src/lib.rs b/library/contracts/safety/src/lib.rs
index 4b6f4528a56f1..9fe852a622de3 100644
--- a/library/contracts/safety/src/lib.rs
+++ b/library/contracts/safety/src/lib.rs
@@ -21,5 +21,5 @@ pub fn requires(attr: TokenStream, item: TokenStream) -> TokenStream {
 #[proc_macro_error]
 #[proc_macro_attribute]
 pub fn ensures(attr: TokenStream, item: TokenStream) -> TokenStream {
-    tool::requires(attr, item)
+    tool::ensures(attr, item)
 }