summaryrefslogtreecommitdiff
path: root/src/ExtractToFStar.ml
diff options
context:
space:
mode:
Diffstat (limited to 'src/ExtractToFStar.ml')
-rw-r--r--src/ExtractToFStar.ml4
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;