diff --git a/test_dockerfiles/docker-compose.yml b/test_dockerfiles/docker-compose.yml
index 0ae0241d5e4a596b2aa54ae625268f431f8a52ef..850fb650b0d740a229f8d2311b4bf60afaad59ba 100644
--- a/test_dockerfiles/docker-compose.yml
+++ b/test_dockerfiles/docker-compose.yml
@@ -1,5 +1,3 @@
-version: '3.8'
-
 services:
   kafka:
     build:
@@ -8,7 +6,7 @@ services:
     ports:
       - "9092:9092"
     environment:
-      KAFKA_ADVERTISED_LISTENERS: PLAINTEXT://localhost:9092
+      KAFKA_ADVERTISED_LISTENERS: PLAINTEXT://kafka:9092
       KAFKA_LISTENER_SECURITY_PROTOCOL_MAP: PLAINTEXT:PLAINTEXT
       KAFKA_INTER_BROKER_LISTENER_NAME: PLAINTEXT
       KAFKA_LOG_DIRS: /tmp/kafka-logs
@@ -21,8 +19,8 @@ services:
       dockerfile: Dockerfile.producer
     depends_on:
       - kafka
-    ports:
-      - "5000:5000"
+    environment:
+      KAFKA_BOOTSTRAP_SERVERS: "kafka:9092"
     volumes:
       - .:/app
 
@@ -32,8 +30,8 @@ services:
       dockerfile: Dockerfile.consumer
     depends_on:
       - kafka
-    ports:
-      - "5001:5001"
+    environment:
+      KAFKA_BOOTSTRAP_SERVERS: "kafka:9092"
     volumes:
       - .:/app
 
diff --git a/requirements.txt b/test_dockerfiles/requirements.txt
similarity index 100%
rename from requirements.txt
rename to test_dockerfiles/requirements.txt