let get_return_type ~env = match lookup_var_type ~env "return" with (_, Var t) -> t | _ -> assert false