diff --git a/crates/bevy_ecs/compile_fail/tests/ui/query_lifetime_safety.rs b/crates/bevy_ecs/compile_fail/tests/ui/query_lifetime_safety.rs
index a8db25b2235d3..905de0d8140e8 100644
--- a/crates/bevy_ecs/compile_fail/tests/ui/query_lifetime_safety.rs
+++ b/crates/bevy_ecs/compile_fail/tests/ui/query_lifetime_safety.rs
@@ -27,15 +27,15 @@ fn main() {
         }
 
         {
-            let data: &Foo = query.single();
-            let mut data2: Mut<Foo> = query.single_mut();
+            let data: &Foo = query.get_single().unwrap();
+            let mut data2: Mut<Foo> = query.get_single_mut().unwrap();
             //~^ E0502
             assert_eq!(data, &mut *data2); // oops UB
         }
 
         {
-            let mut data2: Mut<Foo> = query.single_mut();
-            let data: &Foo = query.single();
+            let mut data2: Mut<Foo> = query.get_single_mut().unwrap();
+            let data: &Foo = query.get_single().unwrap();
             //~^ E0502
             assert_eq!(data, &mut *data2); // oops UB
         }
diff --git a/crates/bevy_ecs/compile_fail/tests/ui/query_transmute_safety.rs b/crates/bevy_ecs/compile_fail/tests/ui/query_transmute_safety.rs
index 489c81d356bec..5b832c1be6948 100644
--- a/crates/bevy_ecs/compile_fail/tests/ui/query_transmute_safety.rs
+++ b/crates/bevy_ecs/compile_fail/tests/ui/query_transmute_safety.rs
@@ -22,8 +22,8 @@ fn main() {
         let mut query_a = lens_a.query();
         let mut query_b = lens_b.query();
 
-        let a = query_a.single_mut();
-        let b = query_b.single_mut(); // oops 2 mutable references to same Foo
+        let a = query_a.get_single_mut().unwrap();
+        let b = query_b.get_single_mut().unwrap(); // oops 2 mutable references to same Foo
         assert_eq!(*a, *b);
     }
 
@@ -34,8 +34,8 @@ fn main() {
         let mut query_b = lens.query();
         //~^ E0499
 
-        let a = query_a.single_mut();
-        let b = query_b.single_mut(); // oops 2 mutable references to same Foo
+        let a = query_a.get_single_mut().unwrap();
+        let b = query_b.get_single_mut().unwrap(); // oops 2 mutable references to same Foo
         assert_eq!(*a, *b);
     }
 }