diff --git a/Kafka/Producer.php b/Kafka/Producer.php
index 80c52f56..606b02af 100644
--- a/Kafka/Producer.php
+++ b/Kafka/Producer.php
@@ -45,11 +45,11 @@ class Producer extends Component
 		if (!class_exists(Conf::class)) {
 			return;
 		}
-		$this->conf = new Conf();
-		$this->topicConf = new TopicConf();
+		$this->conf = di(Conf::class);
 		$this->conf->setErrorCb(function ($kafka, $err, $reason) {
 			$this->error(sprintf("Kafka error: %s (reason: %s)", rd_kafka_err2str($err), $reason));
 		});
+		$this->topicConf = di(TopicConf::class);
 	}
 
 
@@ -93,7 +93,7 @@ class Producer extends Component
 	 */
 	private function getProducer(): \RdKafka\Producer
 	{
-		return Snowflake::createObject(\RdKafka\Producer::class, [$this->conf]);
+		return Snowflake::getDi()->get(\RdKafka\Producer::class, [$this->conf]);
 	}
 
 
diff --git a/Kafka/Struct.php b/Kafka/Struct.php
index 9dc755eb..9476d943 100644
--- a/Kafka/Struct.php
+++ b/Kafka/Struct.php
@@ -20,7 +20,7 @@ class Struct
 	/**
 	 * Struct constructor.
 	 * @param $topic
-	 * @param $message
+	 * @param Message $message
 	 */
 	public function __construct($topic, Message $message)
 	{
diff --git a/Kafka/TaskContainer.php b/Kafka/TaskContainer.php
index fb1b78ca..f5169751 100644
--- a/Kafka/TaskContainer.php
+++ b/Kafka/TaskContainer.php
@@ -31,11 +31,11 @@ class TaskContainer extends BaseObject
     }
 
 
-    /**
-     * @param $topic
-     * @return mixed|null
-     */
-    public function getConsumer(string $topic)
+	/**
+	 * @param string $topic
+	 * @return mixed
+	 */
+    public function getConsumer(string $topic): mixed
     {
         return $this->_topics[$topic] ?? null;
     }