diff --git a/treatment-service/integration-test/fetch_treatment_it_test.go b/treatment-service/integration-test/fetch_treatment_it_test.go
index 28c0ee1..e47068e 100644
--- a/treatment-service/integration-test/fetch_treatment_it_test.go
+++ b/treatment-service/integration-test/fetch_treatment_it_test.go
@@ -387,6 +387,8 @@ func (suite *TreatmentServiceTestSuite) SetupSuite() {
 			panic(err)
 		}
 		fmt.Println(kafkaInfo.NetworkSettings.Networks)
+		fmt.Println(kafkaInfo.NetworkSettings.Networks["kafka_default"])
+		fmt.Println(kafkaInfo.NetworkSettings.Networks["kafka_default"].IPAddress)
 		fmt.Println(kafkaInfo.NetworkSettings.Ports)
 		fmt.Println(kafkaInfo.State)
 		kafkaHost, err := kafkaContainer.Host(ctx)
@@ -400,7 +402,7 @@ func (suite *TreatmentServiceTestSuite) SetupSuite() {
 		}
 		fmt.Println(kafkaPort)
 
-		os.Setenv("ASSIGNEDTREATMENTLOGGER_KAFKACONFIG_BROKERS", fmt.Sprintf("%s:%s", kafkaHost, kafkaPort.Port()))
+		os.Setenv("ASSIGNEDTREATMENTLOGGER_KAFKACONFIG_BROKERS", fmt.Sprintf("%s:%s", kafkaInfo.NetworkSettings.Networks["kafka_default"].IPAddress, kafkaPort.Port()))
 	}
 	suite.kafka = kafka