Slightly better visited definition

HigherOrderCO · Jun 10, 2024 · 92773fb · 92773fb
1 parent 0f675a8
commit 92773fb
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/ast.rs b/src/ast.rs
@@ -577,9 +577,9 @@ impl Book {
     for (name, _) in self.defs.iter() {
       let def = &mut compiled_book.defs[lookup[name] as usize];
       if !def.safe {
-        stack.push(&name);
-        // Temporarily set as safe so we won't need a separate "visited" set
-        def.safe = true;
+        for next in rev_dependencies[name.as_str()].iter() {
+          stack.push(next);
+        }
       }
     }