make recycle stream consistent for recycle specifications

BioSTEAMDevelopmentGroup · Sep 4, 2023 · 2c466b0 · 2c466b0
1 parent d7a5468
commit 2c466b0
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/biosteam/_system.py b/biosteam/_system.py
@@ -1630,13 +1630,14 @@ def recycle(self, recycle):
         elif isa(recycle, piping.TemporaryStream):
             self.method = 'fixed-point'
             permanent = self.unit_set
-            for unit in permanent:
+            unit_path = self.unit_path
+            for unit in unit_path:
                 if len(unit.outs) != 1: continue
                 stream = unit.outs[0]
                 if stream.sink in permanent:
                     self._recycle = stream
                     return
-            for unit in permanent:
+            for unit in unit_path:
                 self._recycle = unit.outs[0]
         else:
             raise_recycle_type_error(recycle)