diff --git a/src/DependencyInjection/LlmChainExtension.php b/src/DependencyInjection/LlmChainExtension.php
index 1ed4ffb..ae2cce9 100644
--- a/src/DependencyInjection/LlmChainExtension.php
+++ b/src/DependencyInjection/LlmChainExtension.php
@@ -232,7 +232,7 @@ private function processChainConfig(string $name, array $config, ContainerBuilde
         };
         $llmDefinition = new Definition($llmClass);
         if (null !== $version) {
-            $llmDefinition->setArgument('$version', $version);
+            $llmDefinition->setArgument('$name', $version);
         }
         if (0 !== count($options)) {
             $llmDefinition->setArgument('$options', $options);
@@ -440,7 +440,7 @@ private function processEmbedderConfig(int|string $name, array $config, Containe
         };
         $modelDefinition = (new Definition($modelClass));
         if (null !== $version) {
-            $modelDefinition->setArgument('$version', $version);
+            $modelDefinition->setArgument('$name', $version);
         }
         if (0 !== count($options)) {
             $modelDefinition->setArgument('$options', $options);