1 files changed, 9 insertions, 5 deletions
diff --git a/test/lit/passes/simplify-locals-gc-nn.wast b/test/lit/passes/simplify-locals-gc-nn.wast
index df5148d12..32c881de8 100644
--- a/test/lit/passes/simplify-locals-gc-nn.wast
+++ b/test/lit/passes/simplify-locals-gc-nn.wast
@@ -154,7 +154,9 @@
   ;; CHECK-NEXT:  (local $temp ((ref func) nullref))
   ;; CHECK-NEXT:  (if
   ;; CHECK-NEXT:   (i32.const 0)
-  ;; CHECK-NEXT:   (nop)
+  ;; CHECK-NEXT:   (then
+  ;; CHECK-NEXT:    (nop)
+  ;; CHECK-NEXT:   )
   ;; CHECK-NEXT:  )
   ;; CHECK-NEXT: )
   (func $if-return-tuple-nn
@@ -165,10 +167,12 @@
     ;; Instead, we can remove the local.set entirely, as it has no gets.
     (if
       (i32.const 0)
-      (local.set $temp
-        (tuple.make 2
-          (ref.func $if-return-tuple-nn)
-          (ref.null none)
+      (then
+        (local.set $temp
+          (tuple.make 2
+            (ref.func $if-return-tuple-nn)
+            (ref.null none)
+          )
         )
       )
     )