diff --git a/tests/benchmark_decode.py b/tests/benchmark_decode.py
index 3f75010..f9a3493 100644
--- a/tests/benchmark_decode.py
+++ b/tests/benchmark_decode.py
@@ -155,7 +155,7 @@ def _run_benchmark(
         # fakesink path: count frames via identity handoff signal
         probe = pipeline.get_by_name("probe")
         if probe:
-            def on_handoff(elem, buf, pad):
+            def on_handoff(*args):
                 n = frame_count[0]
                 frame_count[0] = n + 1
                 if n >= WARMUP_FRAMES: