diff --git a/crates/oxc_semantic/tests/integration/snapshots/arrow_expressions.snap b/crates/oxc_semantic/tests/integration/snapshots/arrow_expressions.snap
index cb8781ba012b8..17c114584500f 100644
--- a/crates/oxc_semantic/tests/integration/snapshots/arrow_expressions.snap
+++ b/crates/oxc_semantic/tests/integration/snapshots/arrow_expressions.snap
@@ -18,6 +18,7 @@ bb2: {
 
 bb3: {
 	statement
+	return
 }
 
 bb4: {
@@ -26,15 +27,16 @@ bb4: {
 
 bb5: {
 	statement
+	return
 }
 
 digraph {
     0 [ label = "" ]
     1 [ label = "ExpressionStatement\nVariableDeclaration" ]
     2 [ label = "" ]
-    3 [ label = "ExpressionStatement" ]
+    3 [ label = "ExpressionStatement\nreturn" ]
     4 [ label = "" ]
-    5 [ label = "ExpressionStatement" ]
+    5 [ label = "ExpressionStatement\nreturn" ]
     1 -> 0 [ label = "Error(Implicit)" ]
     3 -> 2 [ label = "Error(Implicit)" ]
     1 -> 3 [ label = "NewFunction" ]
diff --git a/crates/oxc_semantic/tests/integration/snapshots/cond_expr_in_arrow_fn.snap b/crates/oxc_semantic/tests/integration/snapshots/cond_expr_in_arrow_fn.snap
index 5e7a79e9d20f1..fbdc2d22d484f 100644
--- a/crates/oxc_semantic/tests/integration/snapshots/cond_expr_in_arrow_fn.snap
+++ b/crates/oxc_semantic/tests/integration/snapshots/cond_expr_in_arrow_fn.snap
@@ -32,7 +32,7 @@ bb6: {
 }
 
 bb7: {
-
+	return
 }
 
 digraph {
@@ -43,7 +43,7 @@ digraph {
     4 [ label = "Condition(CallExpression(a))" ]
     5 [ label = "" ]
     6 [ label = "" ]
-    7 [ label = "" ]
+    7 [ label = "return" ]
     1 -> 0 [ label = "Error(Implicit)" ]
     3 -> 2 [ label = "Error(Implicit)" ]
     1 -> 3 [ label = "NewFunction" ]
diff --git a/crates/oxc_semantic/tests/integration/snapshots/import_as_function.snap b/crates/oxc_semantic/tests/integration/snapshots/import_as_function.snap
index fd80beaa47729..72d55b02dbfe3 100644
--- a/crates/oxc_semantic/tests/integration/snapshots/import_as_function.snap
+++ b/crates/oxc_semantic/tests/integration/snapshots/import_as_function.snap
@@ -17,13 +17,14 @@ bb2: {
 
 bb3: {
 	statement
+	return
 }
 
 digraph {
     0 [ label = "" ]
     1 [ label = "ExpressionStatement" ]
     2 [ label = "" ]
-    3 [ label = "ExpressionStatement" ]
+    3 [ label = "ExpressionStatement\nreturn" ]
     1 -> 0 [ label = "Error(Implicit)" ]
     3 -> 2 [ label = "Error(Implicit)" ]
     1 -> 3 [ label = "NewFunction" ]