diff --git a/clients/python/tests/worker/test_producer.py b/clients/python/tests/worker/test_producer.py
index 43ab3be0..2efb42fb 100644
--- a/clients/python/tests/worker/test_producer.py
+++ b/clients/python/tests/worker/test_producer.py
@@ -26,7 +26,7 @@ def __init__(self, use_simple_futures: bool):
         self.use_simple_futures = use_simple_futures
 
     def produce(
-        self, topic: Topic, payload: KafkaPayload
+        self, destination: Topic | Partition, payload: KafkaPayload
     ) -> ProducerFuture[BrokerValue[KafkaPayload]]:
         future: ProducerFuture[BrokerValue[KafkaPayload]]
         if self.use_simple_futures: