diff --git a/tests/flow_stats.py b/tests/flow_stats.py
index a959102..4e115aa 100644
--- a/tests/flow_stats.py
+++ b/tests/flow_stats.py
@@ -74,8 +74,7 @@
         exp_pkt_arg = pkt
         exp_port = egress_port
 
-    (rcv_port, rcv_pkt, pkt_time) = obj.dataplane.poll(timeout=1, 
-                                                       port_number=exp_port,
+    (rcv_port, rcv_pkt, pkt_time) = obj.dataplane.poll(port_number=exp_port,
                                                        exp_pkt=exp_pkt_arg)
     obj.assertTrue(rcv_pkt is not None,
                    "Packet not received on port " + str(egress_port))