From 7f835bf80a98036d4c44a5e6cbd39e776934c1d8 Mon Sep 17 00:00:00 2001
From: Josh Berdine <jjb@fb.com>
Date: Tue, 12 Jan 2021 04:26:01 -0800
Subject: [PATCH] [sledge] Fix Context.fold_uses_of

Summary:
Context.fold_uses_of should enumerate the transitive subterms of a
term rather than only the immediate subterms.

Reviewed By: jvillard

Differential Revision: D25756553

fbshipit-source-id: a3911d9f5
---
 sledge/src/fol/context.ml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/sledge/src/fol/context.ml b/sledge/src/fol/context.ml
index c493c59e1..9ff27ff80 100644
--- a/sledge/src/fol/context.ml
+++ b/sledge/src/fol/context.ml
@@ -710,7 +710,7 @@ let fold_uses_of r t s ~f =
   let rec fold_ e s ~f =
     let s =
       Iter.fold (Trm.trms e) s ~f:(fun sub s ->
-          if Trm.equal t sub then f e s else s )
+          fold_ ~f sub (if Trm.equal t sub then f e s else s) )
     in
     if is_interpreted e then Iter.fold ~f:(fold_ ~f) (Trm.trms e) s else s
   in