diff options
Diffstat (limited to 'src/ExtractToFStar.ml')
-rw-r--r-- | src/ExtractToFStar.ml | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/ExtractToFStar.ml b/src/ExtractToFStar.ml index 0a06ef73..068448e9 100644 --- a/src/ExtractToFStar.ml +++ b/src/ExtractToFStar.ml @@ -1475,8 +1475,8 @@ let extract_unit_test_if_unit_fun (ctx : extraction_ctx) (fmt : F.formatter) let sg = def.signature in if sg.type_params = [] - && (sg.inputs = [ unit_ty ] || sg.inputs = []) - && sg.outputs = [ mk_result_ty unit_ty ] + && (sg.inputs = [ mk_unit_ty ] || sg.inputs = []) + && sg.outputs = [ mk_result_ty mk_unit_ty ] then ( (* Add a break before *) F.pp_print_break fmt 0 0; |