diff --git a/tests/expected/attr-invariant/check_invariant/check_invariant.rs b/tests/expected/attr-invariant/check_invariant/check_invariant.rs
index 176533e93a5f..97f5c74adbb1 100644
--- a/tests/expected/attr-invariant/check_invariant/check_invariant.rs
+++ b/tests/expected/attr-invariant/check_invariant/check_invariant.rs
@@ -5,6 +5,7 @@
 //! `Arbitrary` and `Invariant` implementations for the annotated type.
 
 extern crate kani;
+use kani::Invariant;
 
 #[kani::invariant(x.is_safe() && y.is_safe())]
 struct Point {