From 634c0df52534335ae39644db8ea3701385d7eca4 Mon Sep 17 00:00:00 2001
From: Nathan Villaescusa <nathan@artie.so>
Date: Fri, 29 Mar 2024 12:21:21 -0700
Subject: [PATCH] Simplify

---
 lib/writer/writer_test.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lib/writer/writer_test.go b/lib/writer/writer_test.go
index 0a8edad5..06ac58fb 100644
--- a/lib/writer/writer_test.go
+++ b/lib/writer/writer_test.go
@@ -79,7 +79,7 @@ func TestWriter_Write(t *testing.T) {
 		// Destination error
 		destination := &mockDestination{emitError: true}
 		writer := New(destination)
-		iterator := lib.NewBatchIterator([][]lib.RawMessage{{{TopicSuffix: "a"}}})
+		iterator := lib.NewSingleBatchIterator([]lib.RawMessage{{TopicSuffix: "a"}})
 		_, err := writer.Write(context.Background(), iterator)
 		assert.ErrorContains(t, err, "failed to write messages: test write-raw-messages error")
 		assert.Empty(t, destination.messages)