diff --git a/test-data/simple/simple.expected.ron b/test-data/simple/simple.expected.ron
index baad5a38d3..33281f1bbb 100644
--- a/test-data/simple/simple.expected.ron
+++ b/test-data/simple/simple.expected.ron
@@ -99,6 +99,9 @@
                     pointer: 2,
                     value: 6,
                 ),
+                Return(
+                    value: None,
+                ),
             ],
         ),
     ],
diff --git a/test-data/simple/simple.vert b/test-data/simple/simple.vert
index 507039ffda..27c9f04061 100644
--- a/test-data/simple/simple.vert
+++ b/test-data/simple/simple.vert
@@ -6,4 +6,5 @@ layout(location = 0) out vec4 o_pos;
 void main() {
     float w = 1.0;
     o_pos = vec4(a_pos, 0.0, w);
+    return;
 }
diff --git a/test-data/simple/simple.wgsl b/test-data/simple/simple.wgsl
index bac4c31f07..85eef76145 100644
--- a/test-data/simple/simple.wgsl
+++ b/test-data/simple/simple.wgsl
@@ -5,5 +5,6 @@
 fn main() -> void {
   var w: f32 = 1.0;
   o_pos = vec4<f32>(a_pos, 0.0, w);
+  return;
 }
 entry_point vertex as "main" = main;