diff --git a/treatment-service/integration-test/fetch_treatment_it_test.go b/treatment-service/integration-test/fetch_treatment_it_test.go
index eef9790..847d9eb 100644
--- a/treatment-service/integration-test/fetch_treatment_it_test.go
+++ b/treatment-service/integration-test/fetch_treatment_it_test.go
@@ -370,7 +370,19 @@ func (suite *TreatmentServiceTestSuite) SetupSuite() {
 		panic(err)
 	}
 	fmt.Println(brokers)
-	pinger, err := probing.NewPinger(brokers[0])
+
+	networks, err := kafkaContainer.Networks(ctx)
+	if err != nil {
+		panic(err)
+	}
+	fmt.Println(networks)
+
+	port, err := kafkaContainer.MappedPort(ctx, "9093/tcp")
+	if err != nil {
+		panic(err)
+	}
+
+	pinger, err := probing.NewPinger(fmt.Sprintf("127.0.0.1:%s", port))
 	if err != nil {
 		panic(err)
 	}