diff --git a/flow/e2e/main_test.go b/flow/e2e/main_test.go
index 62bfd17fe0..b0e5146273 100644
--- a/flow/e2e/main_test.go
+++ b/flow/e2e/main_test.go
@@ -1,13 +1,10 @@
 package e2e
 
 import (
-	"os"
-	"testing"
-
 	"github.com/PeerDB-io/peer-flow/cmd"
 )
 
-func TestMain(m *testing.M) {
+func init() {
 	peerWorker, peerErr := cmd.WorkerMain(&cmd.WorkerOptions{
 		TemporalHostPort:  "localhost:7233",
 		EnableProfiling:   false,
@@ -33,6 +30,4 @@ func TestMain(m *testing.M) {
 	} else if err := snapWorker.Start(); err != nil {
 		panic(err)
 	}
-
-	os.Exit(m.Run())
 }