diff --git a/shared/src/main/scala/kinesis4cats/producer/Producer.scala b/shared/src/main/scala/kinesis4cats/producer/Producer.scala
index 4ee305ab..35fe0e7b 100644
--- a/shared/src/main/scala/kinesis4cats/producer/Producer.scala
+++ b/shared/src/main/scala/kinesis4cats/producer/Producer.scala
@@ -142,7 +142,7 @@ abstract class Producer[F[_], PutReq, PutRes] private[kinesis4cats] (
               .parTraverseN(config.shardParallelism) { shardBatch =>
                 putImpl(asPutRequest(shardBatch))
                   .map(resp =>
-                    failedRecords(records, resp)
+                    failedRecords(shardBatch, resp)
                       .map(Producer.Result.putFailures[PutRes])
                       .fold(
                         Producer.Result.success(resp)