diff --git a/examples/graph_to_svg.py b/examples/graph_to_svg.py
index 8a57550..6863d65 100644
--- a/examples/graph_to_svg.py
+++ b/examples/graph_to_svg.py
@@ -8,10 +8,15 @@
 
 parser = argparse.ArgumentParser()
 parser.add_argument("workflow", nargs="?", default="acyclic1")
+parser.add_argument("--test", action="store_true", default=False)
 args = parser.parse_args()
 
-graph_description, _ = get_graph(args.workflow)
-graph = load_graph(graph_description)
-output_path = Path("examples") / f"{args.workflow}.svg"
+if args.test or args.workflow == "acyclic1":
+    graph_to_load, _ = get_graph(args.workflow)
+else:
+    graph_to_load = args.workflow
+
+graph = load_graph(graph_to_load)
+output_path = Path("examples") / f"{Path(args.workflow).name}.svg"
 graph_to_svg(graph, output_path)
 print(f"Wrote {output_path}")