diff --git a/docker/runtime/universe/Dockerfile b/docker/runtime/universe/Dockerfile
index b6df23405..12b4dd0e3 100644
--- a/docker/runtime/universe/Dockerfile
+++ b/docker/runtime/universe/Dockerfile
@@ -2,7 +2,7 @@ ARG BASE_IMAGE="nightly"
 FROM cloudtik/spark-runtime:"$BASE_IMAGE"
 
 ARG ZOOKEEPER_VERSION=3.7.1
-ARG KAFKA_VERSION=3.2.3
+ARG KAFKA_VERSION=3.5.1
 ARG KAFKA_SCALA_VERSION=2.13
 ARG PRESTO_VERSION=0.276
 
diff --git a/python/cloudtik/runtime/kafka/scripts/install.sh b/python/cloudtik/runtime/kafka/scripts/install.sh
index 404f1dde2..a067dd8df 100644
--- a/python/cloudtik/runtime/kafka/scripts/install.sh
+++ b/python/cloudtik/runtime/kafka/scripts/install.sh
@@ -7,7 +7,7 @@ ROOT_DIR="$(dirname "$(dirname "$BIN_DIR")")"
 args=$(getopt -a -o h:: -l head:: -- "$@")
 eval set -- "${args}"
 
-export KAFKA_VERSION=3.2.3
+export KAFKA_VERSION=3.5.1
 export KAFKA_SCALA_VERSION=2.13
 
 export USER_HOME=/home/$(whoami)