Fix compilation of 04-rule-free-variables.rego

1 files changed, 13 insertions, 13 deletions

diff --git a/compiler/passes/compile-constraints.scm b/compiler/passes/compile-constraints.scm
index 80de0fc..81c8d01 100644
--- a/compiler/passes/compile-constraints.scm
+++ b/compiler/passes/compile-constraints.scm
@@ -68,19 +68,19 @@
            (is-local-assignment? (first xs)))
       (let ((assignments (take-while is-local-assignment? xs))
             (body        (drop-while is-local-assignment? xs)))
-        (make <lean-intro-locals>
-          #:identifiers
-          (map (match-lambda
-                 ((($ <reference> (($ <variable-ref> "assign"))) left right)
-                  `(,left ,right)))
-               assignments)
-          #:body
-          (make <lean-conjunction> #:clauses body)))
+        `(,(make <lean-intro-locals>
+             #:identifiers
+             (map (match-lambda
+                    ((($ <reference> (($ <variable-ref> "assign"))) left right)
+                     `(,left ,right)))
+                  assignments)
+             #:body
+             (make <lean-conjunction> #:clauses body))))
       xs))
 
 (define (process-all form)
-  (map process-local-extent
-       (collapse-extents (partition-on form is-local-assignment?))))
+  (fold-right append '() (map process-local-extent
+       (collapse-extents (partition-on form is-local-assignment?)))))
 
 (define (compile-constraints policy)
   (policy-map
@@ -101,7 +101,7 @@
               #:left left
               #:right right)))
        ((xs ...)
-        (if (any is-local-assignment? form)
-            (process-all form)
-            form))
+        (if (any is-local-assignment? xs)
+            (process-all xs)
+            xs))
        (anything-else anything-else)))))