diff --git a/component/src/test/java/org/wso2/extension/siddhi/io/kafka/source/KafkaSourceTestCase.java b/component/src/test/java/org/wso2/extension/siddhi/io/kafka/source/KafkaSourceTestCase.java
index 47fba0e2..98d71f25 100644
--- a/component/src/test/java/org/wso2/extension/siddhi/io/kafka/source/KafkaSourceTestCase.java
+++ b/component/src/test/java/org/wso2/extension/siddhi/io/kafka/source/KafkaSourceTestCase.java
@@ -697,10 +697,11 @@ public void receive(Event[] events) {
             List<Long> expectedValues = new ArrayList<>(2);
             expectedValues.add(0L);
             expectedValues.add(0L);
+            Thread.sleep(3000);
+            AssertJUnit.assertEquals(2, count);
             AssertJUnit.assertEquals("Kafka Source expected input not received", expectedNames,
                     receivedEventNameList);
             AssertJUnit.assertEquals("Kafka Source expected input not received", expectedValues, receivedValueList);
-            AssertJUnit.assertEquals(2, count);
             KafkaTestUtil.deleteTopic(topics);
             siddhiAppRuntime.shutdown();
         } catch (ZkTimeoutException ex) {