diff --git a/examples/sim/dataflow.py b/examples/sim/dataflow.py
index 42863f920..66021ceb6 100644
--- a/examples/sim/dataflow.py
+++ b/examples/sim/dataflow.py
@@ -23,7 +23,7 @@ def main():
 	g.add_connection(source, sink)
 	comp = CompositeActor(g)
 	def end_simulation(s):
-		s.interrupt = source.actor.done
+		s.interrupt = source.actor.token_exchanger.done
 	fragment = comp.get_fragment() + Fragment(sim=[end_simulation])
 	sim = Simulator(fragment, Runner())
 	sim.run()