diff --git a/tests/src/test/java/io/streamnative/pulsar/handlers/kop/KafkaRequestHandlerTest.java b/tests/src/test/java/io/streamnative/pulsar/handlers/kop/KafkaRequestHandlerTest.java
index d689a52d1c..d82578b5d0 100644
--- a/tests/src/test/java/io/streamnative/pulsar/handlers/kop/KafkaRequestHandlerTest.java
+++ b/tests/src/test/java/io/streamnative/pulsar/handlers/kop/KafkaRequestHandlerTest.java
@@ -61,6 +61,7 @@
 import org.apache.kafka.common.TopicPartition;
 import org.apache.kafka.common.config.ConfigResource;
 import org.apache.kafka.common.errors.TimeoutException;
+import org.apache.kafka.common.errors.UnknownServerException;
 import org.apache.kafka.common.errors.UnknownTopicOrPartitionException;
 import org.apache.kafka.common.protocol.ApiKeys;
 import org.apache.kafka.common.protocol.Errors;
@@ -355,8 +356,9 @@ public void testCreateInvalidTopics() {
             createTopicsByKafkaAdmin(kafkaAdmin, topicToNumPartitions);
             fail("create a invalid topic should fail");
         } catch (Exception e) {
-            log.info("Failed to create topics: {}", topicToNumPartitions);
-            assertTrue(e.getCause() instanceof TimeoutException);
+            log.info("Failed to create topics: {} caused by {}", topicToNumPartitions, e.getCause());
+            final Throwable cause = e.getCause();
+            assertTrue(cause instanceof TimeoutException || cause instanceof UnknownServerException);
         }
     }