Fix a small issue with the unit type

1 files changed, 3 insertions, 1 deletions

diff --git a/src/PureMicroPasses.ml b/src/PureMicroPasses.ml
index cbc2409d..1ef4b325 100644
--- a/src/PureMicroPasses.ml
+++ b/src/PureMicroPasses.ml
@@ -633,7 +633,9 @@ let to_monadic (def : fun_def) : fun_def =
       let id, _ = VarId.fresh var_cnt in
       let var = { id; basename = None; ty = unit_ty } in
       let inputs = [ var ] in
-      { def with signature; inputs })
+      let input_lv = mk_typed_lvalue_from_var var None in
+      let inputs_lvs = [ input_lv ] in
+      { def with signature; inputs; inputs_lvs })
     else def
   in
   (* Then the output type *)