diff --git a/test/code/formatPreservation/attributes.test b/test/code/formatPreservation/attributes.test
index 0b87c03..23800a8 100644
--- a/test/code/formatPreservation/attributes.test
+++ b/test/code/formatPreservation/attributes.test
@@ -151,7 +151,7 @@ new #[A] class
 };
 #[A] function () {
 };
-#[A] fn() => 42;
+#[A] fn() => (42);
 -----
 <?php
 
@@ -176,4 +176,4 @@ class X {};
     B,
     C,
 ]
-class X {};
\ No newline at end of file
+class X {};
diff --git a/test/code/prettyPrinter/stmt/attributes.test b/test/code/prettyPrinter/stmt/attributes.test
index c55f7c3..14b6b4e 100644
--- a/test/code/prettyPrinter/stmt/attributes.test
+++ b/test/code/prettyPrinter/stmt/attributes.test
@@ -55,7 +55,7 @@ trait T
 }
 $x = #[A10] function () {
 };
-$y = #[A11] fn() => 0;
+$y = #[A11] fn() => (0);
 new #[A13] class
 {
 };
\ No newline at end of file