diff --git a/tests/integration/inference/test_text_inference.py b/tests/integration/inference/test_text_inference.py
index a3cfce4fd..5410f20c5 100644
--- a/tests/integration/inference/test_text_inference.py
+++ b/tests/integration/inference/test_text_inference.py
@@ -199,8 +199,8 @@ def test_text_completion_structured_output(client_with_models, text_model_id, te
 
     class AnswerFormat(BaseModel):
         name: str
-        year_born: str
-        year_retired: str
+        year_born: int
+        year_retired: int
 
     tc = TestCase(test_case)
 
diff --git a/tests/integration/test_cases/inference/completion.json b/tests/integration/test_cases/inference/completion.json
index 731ceddbc..447087d19 100644
--- a/tests/integration/test_cases/inference/completion.json
+++ b/tests/integration/test_cases/inference/completion.json
@@ -40,8 +40,8 @@
             "user_input": "Michael Jordan was born in 1963. He played basketball for the Chicago Bulls. He retired in 2003.",
             "expected": {
                 "name": "Michael Jordan",
-                "year_born": "1963",
-                "year_retired": "2003"
+                "year_born": 1963,
+                "year_retired": 2003
             }
         }
     },