diff --git a/examples/transmission_main_example.py b/examples/transmission_main_example.py
index 17c79609..b00adbac 100755
--- a/examples/transmission_main_example.py
+++ b/examples/transmission_main_example.py
@@ -178,7 +178,8 @@ if __name__ == '__main__':
     equipment = load_equipment(args.equipment)
     network = load_network(args.filename, equipment, args.names_matching)
 
-    plot_baseline(network)
+    if args.plot:
+        plot_baseline(network)
 
     transceivers = {n.uid: n for n in network.nodes() if isinstance(n, Transceiver)}