diff --git a/backend/tests/integration_tests/test_extraction.py b/backend/tests/integration_tests/test_extraction.py
index fb3b713..86ddb59 100644
--- a/backend/tests/integration_tests/test_extraction.py
+++ b/backend/tests/integration_tests/test_extraction.py
@@ -50,7 +50,6 @@ class Person(BaseModel):
         )
         assert result.status_code == 200, result.text
         response_data = result.json()
-        assert response_data == {}
         assert isinstance(response_data["output"]["data"], list)
 
         # Test with instructions
@@ -79,7 +78,7 @@ class Person(BaseModel):
             json={
                 "input": {
                     "text": text,
-                    "schema": Person(),
+                    "schema": Person.schema(),
                     "instructions": "Redact all names using the characters `######`",
                     "examples": examples,
                 }