diff --git a/test/test_sfg.py b/test/test_sfg.py
index a37caa79850a6f7adac237c68521790424a52a96..b965e80ad014200fdc92e34039238daa7ac4a6f2 100644
--- a/test/test_sfg.py
+++ b/test/test_sfg.py
@@ -1370,7 +1370,7 @@ class TestPrecedenceGraph:
             '[label=add1 shape=square]\n\tcmul1 -> "cmul1.0"\n\tcmul1 '
             '[label=cmul1 shape=square]\n\t"add1.0" -> t1In\n\tt1In '
             '[label=t1 shape=square]\n\tadd1 -> "add1.0"\n\tadd1 '
-            "[label=add1 shape=square]\n}"
+            "[label=add1 shape=square]\n}\n"
         )
 
         assert sfg_simple_filter.precedence_graph().source == res
@@ -1382,7 +1382,7 @@ class TestSFGGraph:
             "digraph {\n\trankdir=LR\n\tin1\n\tin1 -> "
             "add1\n\tout1\n\tt1 -> out1\n\tadd1\n\tcmul1 -> "
             "add1\n\tcmul1\n\tadd1 -> t1\n\tt1 [shape=square]\n\tt1 "
-            "-> cmul1\n}"
+            "-> cmul1\n}\n"
         )
 
         assert sfg_simple_filter.sfg().source == res
@@ -1392,7 +1392,7 @@ class TestSFGGraph:
             "digraph {\n\trankdir=LR\n\tin1\n\tin1 -> add1 "
             "[label=s1]\n\tout1\n\tt1 -> out1 [label=s2]\n\tadd1"
             "\n\tcmul1 -> add1 [label=s3]\n\tcmul1\n\tadd1 -> t1 "
-            "[label=s4]\n\tt1 [shape=square]\n\tt1 -> cmul1 [label=s5]\n}"
+            "[label=s4]\n\tt1 [shape=square]\n\tt1 -> cmul1 [label=s5]\n}\n"
         )
 
         assert sfg_simple_filter.sfg(show_id=True).source == res