Update tests

1 files changed, 9 insertions, 4 deletions

diff --git a/embassy-executor/tests/ui.rs b/embassy-executor/tests/ui.rs
index ed8228e27..c4a1a601c 100644
--- a/embassy-executor/tests/ui.rs
+++ b/embassy-executor/tests/ui.rs
@@ -17,10 +17,15 @@ fn ui() {
     t.compile_fail("tests/ui/nonstatic_struct_elided.rs");
     t.compile_fail("tests/ui/nonstatic_struct_generic.rs");
     t.compile_fail("tests/ui/not_async.rs");
-    // #[cfg(not(feature = "nightly"))] // output differs on stable and nightly
-    // t.compile_fail("tests/ui/bad_return_impl_trait.rs");
-    #[cfg(feature = "nightly")]
-    t.compile_fail("tests/ui/bad_return_impl_trait_nightly.rs");
+    if rustversion::cfg!(stable) {
+        // output is slightly different on nightly
+        t.compile_fail("tests/ui/bad_return_impl_future.rs");
+        t.compile_fail("tests/ui/return_impl_send.rs");
+    }
+    if cfg!(feature = "nightly") {
+        t.compile_fail("tests/ui/bad_return_impl_future_nightly.rs");
+        t.compile_fail("tests/ui/return_impl_send_nightly.rs");
+    }
     t.compile_fail("tests/ui/self_ref.rs");
     t.compile_fail("tests/ui/self.rs");
     t.compile_fail("tests/ui/type_error.rs");