From f434afbee65363d83cacb531596bf5f7b44a7352 Mon Sep 17 00:00:00 2001
From: Bill <sunyinqi0508@gmail.com>
Date: Wed, 5 Oct 2022 04:03:17 +0800
Subject: [PATCH] fixed regression

---
 reconstruct/ast.py | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/reconstruct/ast.py b/reconstruct/ast.py
index 7efdef0..da8e01c 100644
--- a/reconstruct/ast.py
+++ b/reconstruct/ast.py
@@ -628,11 +628,12 @@ class join(ast_node):
 
     def strip_joint_cols(self, cols : Set[ColRef]):
         stripped = type(cols)(cols)
-        for c in stripped:
-            jc = self.get_joint_cols([c])
-            for j in jc:
-                if j != c and j in stripped:
-                    stripped.remove(j)
+        for c in cols:
+            if c in stripped:
+                jc = self.get_joint_cols([c])
+                for j in jc:
+                    if j != c and j in stripped:
+                        stripped.remove(j)
         return stripped
     
     def init(self, _):