diff --git a/docker/docker-compose-without-neo4j.yml b/docker/docker-compose-without-neo4j.yml
index 433426a804..0cedaa25b2 100644
--- a/docker/docker-compose-without-neo4j.yml
+++ b/docker/docker-compose-without-neo4j.yml
@@ -103,7 +103,8 @@ services:
 
   kafka-setup:
     build:
-      context: kafka-setup
+      dockerfile: ./docker/kafka-setup/Dockerfile
+      context: ../
     image: ${DATAHUB_KAFKA_SETUP_IMAGE:-linkedin/datahub-kafka-setup}:${DATAHUB_VERSION:-head}
     env_file: kafka-setup/env/docker.env
     hostname: kafka-setup
diff --git a/docker/docker-compose.yml b/docker/docker-compose.yml
index 025a57d8dd..ec5cf2f856 100644
--- a/docker/docker-compose.yml
+++ b/docker/docker-compose.yml
@@ -119,7 +119,8 @@ services:
   # explicitly wait for this container
   kafka-setup:
     build:
-      context: kafka-setup
+      dockerfile: ./docker/kafka-setup/Dockerfile
+      context: ../
     image: ${DATAHUB_KAFKA_SETUP_IMAGE:-linkedin/datahub-kafka-setup}:${DATAHUB_VERSION:-head}
     env_file: kafka-setup/env/docker.env
     hostname: kafka-setup