diff --git a/tests/queue_size_and_timeout_python/send_data.py b/tests/queue_size_and_timeout_python/send_data.py
index 2ae6c465..a29e1abb 100644
--- a/tests/queue_size_and_timeout_python/send_data.py
+++ b/tests/queue_size_and_timeout_python/send_data.py
@@ -3,11 +3,13 @@ import time
 import pyarrow as pa
 from dora import Node
 
+start = time.time()
+
 
 def main() -> None:
     dora_node = Node()
     i = 0
-    while True:
+    while time.time() - start < 10:
         dora_node.send_output("ts", pa.array([time.perf_counter_ns(), i]))
         i += 1
         # print(f"Sent {i} times", flush=True)