diff --git a/api/tests/integration_tests/model_runtime/__mock/plugin_model.py b/api/tests/integration_tests/model_runtime/__mock/plugin_model.py
index 50913662e2..62f5a659aa 100644
--- a/api/tests/integration_tests/model_runtime/__mock/plugin_model.py
+++ b/api/tests/integration_tests/model_runtime/__mock/plugin_model.py
@@ -227,6 +227,7 @@ class MockModelClass(PluginModelManager):
                             total_price=Decimal(0.0003),
                             currency="USD",
                             latency=0.001,
+                            ttft=0,
                         ),
                     ),
                 )