diff --git a/instrumentation/opentelemetry-instrumentation-confluent-kafka/tests/test_instrumentation.py b/instrumentation/opentelemetry-instrumentation-confluent-kafka/tests/test_instrumentation.py
index f1016e7460..e47d43cdcb 100644
--- a/instrumentation/opentelemetry-instrumentation-confluent-kafka/tests/test_instrumentation.py
+++ b/instrumentation/opentelemetry-instrumentation-confluent-kafka/tests/test_instrumentation.py
@@ -176,10 +176,10 @@ def test_poll(self) -> None:
         )
         span_list = self.memory_exporter.clear()
         consumer = instrumentation.instrument_consumer(consumer)
-        consumer.poll(1)
-        consumer.poll(1)
-        consumer.poll(1)
-        consumer.poll(1)
+        consumer.poll()
+        consumer.poll()
+        consumer.poll()
+        consumer.poll()
         
         span_list = self.memory_exporter.get_finished_spans()
         self._compare_spans(span_list, expected_spans)