Fix Rust cosmetic pass removing trailing units

FStarLang · Jan 8, 2025 · e80c30f · e80c30f
1 parent e111ce5
commit e80c30f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/OptimizeMiniRust.ml b/lib/OptimizeMiniRust.ml
@@ -1116,7 +1116,7 @@ let remove_trailing_unit = object
     let e1 = super#visit_expr () e1 in
     let e2 = super#visit_expr () e2 in
     match e2 with
-    | Unit -> e1
+    | Unit when b.typ = Unit -> e1
     | _ -> Let (b, e1, e2)
 end