diff --git a/gfe_examples/live_io/live_io_vertex.py b/gfe_examples/live_io/live_io_vertex.py
index 42bf4fa9..8d39cc03 100644
--- a/gfe_examples/live_io/live_io_vertex.py
+++ b/gfe_examples/live_io/live_io_vertex.py
@@ -69,8 +69,8 @@ def generate_machine_data_specification(
         spec.switch_write_focus(DataRegions.KEY_DATA)
 
         routing_infos = FecDataView().get_routing_infos()
-        r_info = routing_infos.get_routing_info_from_pre_vertex(
-            placement.vertex, self.__send_partition)
+        r_info = routing_infos.get_single_routing_info_from_pre_vertex(
+            placement.vertex)
         if r_info is None:
             spec.write_value(0)
             spec.write_value(0)