diff --git a/tests/peggy_comparison/conditional_constant_folding.bril b/tests/peggy_comparison/conditional_constant_folding.bril
index 097a394c5..4fa76a71d 100644
--- a/tests/peggy_comparison/conditional_constant_folding.bril
+++ b/tests/peggy_comparison/conditional_constant_folding.bril
@@ -7,16 +7,22 @@
 
 .if:
     res: int = mul four x;
-    ret res;
+    jmp .end;
 
 .else_if_check:
     cond: bool = eq x four;
     br cond .else_if .else;
+    jmp .end;
 
 .else_if:
     res: int = mul five x;
-    ret res;
+    jmp .end;
     
 .else:
-    ret twenty;
+    res: int = id twenty;
+    jmp .end;
+
+.end:
+    print res;
+    ret res;
 }
\ No newline at end of file
diff --git a/tests/peggy_comparison/dead_loop_deletion.bril b/tests/peggy_comparison/dead_loop_deletion.bril
index b7b085717..c3e4bd15b 100644
--- a/tests/peggy_comparison/dead_loop_deletion.bril
+++ b/tests/peggy_comparison/dead_loop_deletion.bril
@@ -1,6 +1,6 @@
 @main: int {
     j: int = const 3;
-    i: int = const 1;
+    i: int = const 5;
     four: int = const 4;
     one: int = const 1;