chore: fix tests
diff --git a/tests/lean/run/grind_pre.lean b/tests/lean/run/grind_pre.lean
index 50b8b7b..4452de7 100644
--- a/tests/lean/run/grind_pre.lean
+++ b/tests/lean/run/grind_pre.lean
@@ -161,6 +161,7 @@
     [prop] r
   [cases] Case analyses
     [cases] [1/2]: p = r
+      [cases] source: Initial goal
 -/
 #guard_msgs (error) in
 example (a : α) (p q r : Prop) : (h₁ : HEq p a) → (h₂ : HEq q a) → (h₃ : p = r) → False := by
diff --git a/tests/lean/run/grind_split_issue.lean b/tests/lean/run/grind_split_issue.lean
index 37ff84b..ae16f47 100644
--- a/tests/lean/run/grind_split_issue.lean
+++ b/tests/lean/run/grind_split_issue.lean
@@ -25,6 +25,7 @@
     [eqc] {s, 0}
   [cases] Case analyses
     [cases] [1/2]: X c 0
+      [cases] source: Initial goal
   [cutsat] Assignment satisfying linear constraints
     [assign] c := 1
     [assign] s := 0