diff --git a/datahub-upgrade/build.gradle b/datahub-upgrade/build.gradle
index 304bf3a67a5b2..f64886953fe22 100644
--- a/datahub-upgrade/build.gradle
+++ b/datahub-upgrade/build.gradle
@@ -55,7 +55,7 @@ dependencies {
   // mock internal schema registry
   implementation externalDependency.kafkaAvroSerde
   implementation externalDependency.kafkaAvroSerializer
-  implementation "org.apache.kafka:kafka_2.12:3.7.0"
+  implementation "org.apache.kafka:kafka_2.12:3.7.1"
 
   implementation externalDependency.slf4jApi
   compileOnly externalDependency.lombok
diff --git a/docker/kafka-setup/Dockerfile b/docker/kafka-setup/Dockerfile
index eb7ffa1e1590b..1ba717641c2a2 100644
--- a/docker/kafka-setup/Dockerfile
+++ b/docker/kafka-setup/Dockerfile
@@ -22,8 +22,8 @@ ARG ALPINE_REPO_URL
 ARG APACHE_DOWNLOAD_URL
 ARG GITHUB_REPO_URL
 
-ENV KAFKA_VERSION 3.7.0
-ENV SCALA_VERSION 2.13
+ENV KAFKA_VERSION=3.7.1
+ENV SCALA_VERSION=2.13
 
 LABEL name="kafka" version=${KAFKA_VERSION}