diff --git a/sf-writer/test/manual/test_sf_writer.cpp b/sf-writer/test/manual/test_sf_writer.cpp
index 57102b7..c992586 100644
--- a/sf-writer/test/manual/test_sf_writer.cpp
+++ b/sf-writer/test/manual/test_sf_writer.cpp
@@ -44,7 +44,7 @@ void receive_replay(
 
             cout << "Received " << image_metadata->pulse_id << endl;
 
-            receiver.get_next_image(
+            receiver.get_next_batch(
                     current_pulse_id, image_metadata, image_buffer);
 
             if (image_metadata->pulse_id != current_pulse_id) {
diff --git a/sf-writer/test/test_WriterZmqReceiver.cpp b/sf-writer/test/test_WriterZmqReceiver.cpp
index 847352e..90996c2 100644
--- a/sf-writer/test/test_WriterZmqReceiver.cpp
+++ b/sf-writer/test/test_WriterZmqReceiver.cpp
@@ -70,7 +70,7 @@ TEST(WriterZmqReceiver, basic_test)
                  0);
     }
 
-    receiver.get_next_image(pulse_id, &image_metadata, image_buffer.get());
+    receiver.get_next_batch(pulse_id, &image_metadata, image_buffer.get());
     EXPECT_EQ(pulse_id, image_metadata.pulse_id);
     EXPECT_EQ(image_metadata.is_good_frame, 1);
     EXPECT_EQ(image_metadata.daq_rec, 4);