diff --git a/io/transmitPhasestream_test.go b/io/transmitPhasestream_test.go
index 5e70ca652d2ce9198ef3c540e1f64903054726f6..e4438faf864d44ff37559292460be8f274724604 100644
--- a/io/transmitPhasestream_test.go
+++ b/io/transmitPhasestream_test.go
@@ -93,7 +93,7 @@ func TestStreamPostPhase(t *testing.T) {
 	// receive the mockBatch into the mock stream 'buffer'
 	mockStreamServer := MockTransmitStream{batch: mockBatch}
 
-	_, err := StreamPostPhase(mockPhase, uint32(batchSize), mockStreamServer)
+	_, _, err := StreamPostPhase(mockPhase, uint32(batchSize), mockStreamServer)
 
 	if err != nil {
 		t.Errorf("StreamPostPhase: Unexpected error returned: %+v", err)