Fixed computation of arity for top-level inerts

[fireball-separation.git] / ocaml / parser.ml

diff --git a/ocaml/parser.ml b/ocaml/parser.ml
index 8ba2fbd7ae9a3f6770146f83ded59e5e151d9dc7..79650955de2f5c5f7fa73606dee2e68a9613f73d 100644 (file)
--- a/ocaml/parser.ml
+++ b/ocaml/parser.ml
@@ -172,7 +172,7 @@ let problem_of_string s =
  (* Fix arity *)\r
  let open Num in\r
  let rec aux lev : nf -> nf = function\r
- | `I((n,_), args) -> `I((n,1 + Listx.length args), Listx.map (fun t -> aux lev t) args)\r
+ | `I((n,_), args) -> `I((n,(if lev = 0 then 0 else 1) + Listx.length args), Listx.map (aux lev) args)\r
  | `Var(n,_) -> fix lev n\r
  | `Lam(_,t) -> `Lam (true, aux (lev+1) t)\r
  | `Match _ | `N _ -> assert false in\r