simplify

1 files changed, 4 insertions, 6 deletions

diff --git a/meliaereader/reader.py b/meliaereader/reader.py
index 876df375..f8a4b925 100644
--- a/meliaereader/reader.py
+++ b/meliaereader/reader.py
@@ -80,25 +80,23 @@ class MeliaeReader(object):
                 'computing closures for {} objs\n'.format(len(growing_refs)))
             growing2 = set()
             for ref in growing_refs:
-                growing2.update(self.add_to_closure(ref))
+                if self.add_to_closure(ref):
+                    growing2.add(ref)
             growing_refs = growing2
 
         assert set(self._objs.keys()) == set(self._closures.keys())
 
     def add_to_closure(self, ref):
-        grown = set()
+        grown = False
         closure = self._closures[ref]
         children = [self.get_object(r) for r in closure]
         for child in children:
             delta = set(child['refs']).difference(closure)
             if delta:
-                add = False
                 for r in delta:
                     if r in self:
                         closure.add(r)
-                        add = True
-                if add:
-                    grown.add(child['address'])
+                        grown = True
         return grown
 
     def _simple_get_closure(self, ref):  # pragma: no cover