diff --git a/src/ai-bundle/src/Profiler/TraceablePlatform.php b/src/ai-bundle/src/Profiler/TraceablePlatform.php
index 4d49f4dae..e8d0e01ee 100644
--- a/src/ai-bundle/src/Profiler/TraceablePlatform.php
+++ b/src/ai-bundle/src/Profiler/TraceablePlatform.php
@@ -23,7 +23,7 @@
  * @author Christopher Hertel <mail@christopher-hertel.de>
  *
  * @phpstan-type PlatformCallData array{
- *     model: Model,
+ *     model: string,
  *     input: array<mixed>|string|object,
  *     options: array<string, mixed>,
  *     result: ResultPromise,