diff --git a/src/KafkaHandler.py b/src/KafkaHandler.py
index bb16b6b..002ae89 100644
--- a/src/KafkaHandler.py
+++ b/src/KafkaHandler.py
@@ -13,6 +13,7 @@ def __init__(self, backup_file, hosts_list, topic, batch_size):
         logging.Handler.__init__(self)
         # Backup log file for errors
         self.fail_fh = open(backup_file, 'w')
+        self.batch_size = batch_size
         kafka_client = KafkaClient(hosts_list)
         topic = kafka_client.topics[topic]
         self.key = bytes(str(socket.gethostname()))