Fix lint-unsafe-code test from #22542

1 files changed, 5 insertions, 2 deletions

diff --git a/src/test/compile-fail/lint-unsafe-code.rs b/src/test/compile-fail/lint-unsafe-code.rs
index 486e7142f27..7b17d887757 100644
--- a/src/test/compile-fail/lint-unsafe-code.rs
+++ b/src/test/compile-fail/lint-unsafe-code.rs
@@ -12,13 +12,16 @@
 #![allow(dead_code)]
 #![deny(unsafe_code)]
 
+use std::marker::PhantomFn;
+
 struct Bar;
 
 #[allow(unsafe_code)]
 mod allowed_unsafe {
+    use std::marker::PhantomFn;
     fn allowed() { unsafe {} }
     unsafe fn also_allowed() {}
-    unsafe trait AllowedUnsafe {}
+    unsafe trait AllowedUnsafe : PhantomFn<Self> {}
     unsafe impl AllowedUnsafe for super::Bar {}
 }
 
@@ -29,7 +32,7 @@ macro_rules! unsafe_in_macro {
 }
 
 unsafe fn baz() {} //~ ERROR: declaration of an `unsafe` function
-unsafe trait Foo {} //~ ERROR: declaration of an `unsafe` trait
+unsafe trait Foo : PhantomFn<Self> {} //~ ERROR: declaration of an `unsafe` trait
 unsafe impl Foo for Bar {} //~ ERROR: implementation of an `unsafe` trait
 
 trait Baz {