diff --git a/dd-java-agent/instrumentation/kafka-clients-0.11/src/main/java/datadog/trace/instrumentation/kafka_clients/KafkaConsumerInstrumentation.java b/dd-java-agent/instrumentation/kafka-clients-0.11/src/main/java/datadog/trace/instrumentation/kafka_clients/KafkaConsumerInstrumentation.java
index 9653a61b0b3..d79e8a084b6 100644
--- a/dd-java-agent/instrumentation/kafka-clients-0.11/src/main/java/datadog/trace/instrumentation/kafka_clients/KafkaConsumerInstrumentation.java
+++ b/dd-java-agent/instrumentation/kafka-clients-0.11/src/main/java/datadog/trace/instrumentation/kafka_clients/KafkaConsumerInstrumentation.java
@@ -52,7 +52,9 @@ public Map<? extends ElementMatcher<? super MethodDescription>, String> transfor
         isMethod()
             .and(isPublic())
             .and(named("records"))
-            .and(takesArgument(0, String.class))
+            .and(
+                takesArgument(0, String.class)
+                    .or(takesArgument(0, named("org.apache.kafka.common.TopicPartition"))))
             .and(returns(Iterable.class)),
         IterableAdvice.class.getName());
     transformers.put(