diff --git a/src/Clients/Consumer/KafkaConsumer.php b/src/Clients/Consumer/KafkaConsumer.php
index 74c4b34..9c0bc00 100644
--- a/src/Clients/Consumer/KafkaConsumer.php
+++ b/src/Clients/Consumer/KafkaConsumer.php
@@ -83,9 +83,9 @@ function (RdKafkaConsumer $kafka, int $err, array|null $partitions = null): void
     }
 
     /**
-     * @param callable(Message) : void $onSuccess
-     * @param callable() : void        $onPartitionEof
-     * @param callable() : void        $onTimedOut
+     * @param callable(Message):void $onSuccess
+     * @param (callable():void)|null $onPartitionEof
+     * @param (callable():void)|null $onTimedOut
      */
     public function start(
         int $timeoutMs,
@@ -97,8 +97,8 @@ public function start(
     }
 
     /**
-     * @param callable(Message) : void         $processRecord
-     * @param callable(ConsumerRecords) : void $onBatchProcessed
+     * @param (callable(Message):void)|null         $processRecord
+     * @param (callable(ConsumerRecords):void)|null $onBatchProcessed
      */
     public function startBatch(
         int $maxBatchSize,
@@ -150,9 +150,9 @@ public function shutdown(): void
     }
 
     /**
-     * @param callable(Message): void $onSuccess
-     * @param callable() : void       $onPartitionEof
-     * @param callable() : void       $onTimedOut
+     * @param callable(Message):void $onSuccess
+     * @param (callable():void)|null       $onPartitionEof
+     * @param (callable():void)|null       $onTimedOut
      */
     private function doStart(
         int $timeoutMs,