From 24f67da9c263f7ef5fbc5debbab931fdfeda82de Mon Sep 17 00:00:00 2001
From: Eric Meisel <eric.steven.meisel@gmail.com>
Date: Thu, 9 Nov 2023 12:38:10 -0600
Subject: [PATCH] Use shardBatch instead of records for failedRecords (#248)

---
 shared/src/main/scala/kinesis4cats/producer/Producer.scala | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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)