From cc1bfea6f02c0ee9c495a81eaf5ea12b0bef41c0 Mon Sep 17 00:00:00 2001
From: Justin Ethier <justin.ethier@gmail.com>
Date: Wed, 5 Sep 2018 12:51:21 -0400
Subject: [PATCH] Update (free-vars) to include AST support

---
 scheme/cyclone/transforms.sld | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/scheme/cyclone/transforms.sld b/scheme/cyclone/transforms.sld
index b1a799d6..b2732f4e 100644
--- a/scheme/cyclone/transforms.sld
+++ b/scheme/cyclone/transforms.sld
@@ -630,6 +630,9 @@
   (define (search exp)
     (cond
       ; Core forms:
+      ((ast:lambda? exp)
+        (difference (reduce union (map search (ast:lambda-body exp)) '())
+                    (ast:lambda-formals->list exp)))
       ((const? exp)    '())
       ((prim? exp)     '())    
       ((quote? exp)    '())