diff --git a/flow/e2e/eventhub/eventhub.go b/flow/e2e/eventhub/eventhub.go
index acc0084607..153051b06b 100644
--- a/flow/e2e/eventhub/eventhub.go
+++ b/flow/e2e/eventhub/eventhub.go
@@ -44,7 +44,7 @@ func ConsumeAllMessages(
 		return nil, fmt.Errorf("failed to create eventhub client: %w", err)
 	}
 
-	var messages []string
+	messages := make([]string, 0, expectedNum)
 	partitionClient, err := ehClient.NewPartitionClient("0", nil)
 	if err != nil {
 		return nil, fmt.Errorf("failed to create partition client: %w", err)
diff --git a/flow/e2e/eventhub/peer_flow_eh_test.go b/flow/e2e/eventhub/peer_flow_eh_test.go
index 20b0098fc8..ab49697b82 100644
--- a/flow/e2e/eventhub/peer_flow_eh_test.go
+++ b/flow/e2e/eventhub/peer_flow_eh_test.go
@@ -10,14 +10,15 @@ import (
 	"testing"
 	"time"
 
+	"github.com/jackc/pgx/v5"
+	"github.com/stretchr/testify/require"
+
 	connpostgres "github.com/PeerDB-io/peer-flow/connectors/postgres"
 	"github.com/PeerDB-io/peer-flow/e2e"
 	"github.com/PeerDB-io/peer-flow/e2eshared"
 	"github.com/PeerDB-io/peer-flow/generated/protos"
 	"github.com/PeerDB-io/peer-flow/shared"
 	peerflow "github.com/PeerDB-io/peer-flow/workflows"
-	"github.com/jackc/pgx/v5"
-	"github.com/stretchr/testify/require"
 )
 
 type EventhubsSuite struct {