diff --git a/tests/load.py b/tests/load.py
index d168a17..7b18252 100644
--- a/tests/load.py
+++ b/tests/load.py
@@ -87,7 +87,7 @@
         self.controller.pkt_in_filter_limit = 10
 
         pkt = simple_tcp_packet()
-        match = parse.packet_to_flow_match(pkt)
+        match = packet_to_flow_match(self, pkt)
         match.wildcards &= ~ofp.OFPFW_IN_PORT
         match.in_port = lb_port
         act = action.action_output()