diff --git a/examples/dataflow.py b/examples/dataflow.py
index 5c3d5fc0e..f6e786778 100644
--- a/examples/dataflow.py
+++ b/examples/dataflow.py
@@ -1,5 +1,4 @@
 import sys
-import networkx as nx
 
 from migen.fhdl import verilog
 from migen.flow.ala import *
@@ -19,5 +18,6 @@ print(verilog.convert(frag))
 
 if len(sys.argv) > 1 and sys.argv[1] == "draw":
 	import matplotlib.pyplot as plt
+	import networkx as nx
 	nx.draw(g)
 	plt.show()