diff --git a/build.gradle b/build.gradle
index 4197494..c30b935 100644
--- a/build.gradle
+++ b/build.gradle
@@ -37,7 +37,7 @@ task wrapper(type: Wrapper) {
 }
 
 dependencies {
-    compile 'org.apache.kafka:kafka-clients:2.0.0'
+    compile 'org.apache.kafka:kafka-clients:2.0.1'
     compile 'org.slf4j:slf4j-api:1.7.25'
     compile 'org.lz4:lz4-java:1.4.1'
     compile 'org.xerial.snappy:snappy-java:1.1.7.1'
diff --git a/kafka_test_setup.sh b/kafka_test_setup.sh
index a0065ec..5f2e4dd 100755
--- a/kafka_test_setup.sh
+++ b/kafka_test_setup.sh
@@ -5,7 +5,7 @@ set -ex
 if [ -n "${KAFKA_VERSION+1}" ]; then
   echo "KAFKA_VERSION is $KAFKA_VERSION"
 else
-   KAFKA_VERSION=1.1.1
+   KAFKA_VERSION=2.0.1
 fi
 
 echo "Downloading Kafka version $KAFKA_VERSION"