diff --git a/src/ai-bundle/src/AiBundle.php b/src/ai-bundle/src/AiBundle.php
index fe6883f8d..1eff064cc 100644
--- a/src/ai-bundle/src/AiBundle.php
+++ b/src/ai-bundle/src/AiBundle.php
@@ -248,7 +248,7 @@ private function processPlatformConfig(string $type, array $platform, ContainerB
                         $config['api_key'],
                         new Reference($config['http_client'], ContainerInterface::NULL_ON_INVALID_REFERENCE),
                         new Reference('ai.platform.model_catalog.azure.openai'),
-                        new Reference('ai.platform.contract.azure.openai'),
+                        new Reference('ai.platform.contract.openai'),
                     ])
                     ->addTag('ai.platform');