From 3380e0eed7b053e47d0515a3988fc5e70843f751 Mon Sep 17 00:00:00 2001
From: Max Brunsfeld <maxbrunsfeld@gmail.com>
Date: Tue, 31 Oct 2023 15:59:37 -0700
Subject: [PATCH] Fix hang when printing dot graphs from the CLI

---
 cli/src/parse.rs | 2 ++
 cli/src/test.rs  | 4 +++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/cli/src/parse.rs b/cli/src/parse.rs
index 99d28a74..14c4499d 100644
--- a/cli/src/parse.rs
+++ b/cli/src/parse.rs
@@ -104,6 +104,8 @@ pub fn parse_file_at_path(parser: &mut Parser, opts: ParseFileOptions) -> Result
         _ => parser.parse(&source_code, None),
     };
 
+    parser.stop_printing_dot_graphs();
+
     let stdout = io::stdout();
     let mut stdout = stdout.lock();
 
diff --git a/cli/src/test.rs b/cli/src/test.rs
index 80643aef..2a452b73 100644
--- a/cli/src/test.rs
+++ b/cli/src/test.rs
@@ -90,6 +90,8 @@ pub fn run_tests_at_path(
         &mut corrected_entries,
     )?;
 
+    parser.stop_printing_dot_graphs();
+
     if failures.len() > 0 {
         println!("");
 
@@ -719,7 +721,7 @@ code
 ---
 
 ; Line start comment
-(a 
+(a
 ; ignore this
     (b)
     ; also ignore this