diff --git a/examples/dataflow/structuring.py b/examples/dataflow/structuring.py
index 0525e2337..88c87a394 100644
--- a/examples/dataflow/structuring.py
+++ b/examples/dataflow/structuring.py
@@ -1,3 +1,5 @@
+from itertools import count
+
 import networkx as nx
 import matplotlib.pyplot as plt
 
@@ -11,7 +13,7 @@ from migen.flow import perftools
 pack_factor = 5
 
 def source_gen():
-	for i in range(180):
+	for i in count(0):
 		yield Token("source", {"value": i})
 
 def sink_gen():