diff --git a/parallel-consumer-core/src/main/java/io/confluent/parallelconsumer/RecordProcessor.java b/parallel-consumer-core/src/main/java/io/confluent/parallelconsumer/RecordProcessor.java
index f1c156e45..b7a357065 100644
--- a/parallel-consumer-core/src/main/java/io/confluent/parallelconsumer/RecordProcessor.java
+++ b/parallel-consumer-core/src/main/java/io/confluent/parallelconsumer/RecordProcessor.java
@@ -30,7 +30,7 @@ interface RecordConsumer<K, V> extends java.util.function.Consumer<PollContext<K
     }
 
     @FunctionalInterface
-    interface RecordFunction<T, R> extends java.util.function.Function<T, R> {
+    interface RecordFunction<K, V, R> extends java.util.function.Function<PollContext<K, V>, R> {
 
 
         /**
@@ -40,7 +40,7 @@ interface RecordFunction<T, R> extends java.util.function.Function<T, R> {
          * @return the function result
          */
         @Override
-        R apply(T t);
+        R apply(PollContext t);
 
     }
 }