diff --git a/tests/integration/cloud/rest_contracts/unit_test_mocks/conftest.py b/tests/integration/cloud/rest_contracts/unit_test_mocks/conftest.py
index 5efced19442a..0bff2545fb6f 100644
--- a/tests/integration/cloud/rest_contracts/unit_test_mocks/conftest.py
+++ b/tests/integration/cloud/rest_contracts/unit_test_mocks/conftest.py
@@ -51,7 +51,7 @@ def _reify_pact_body(
 
 def _get_mock_response_from_pact_response_body(
     status_code: int,
-    pact_response_body: dict,
+    pact_response_body: PactBody,
 ) -> requests.Response:
     response_body: JsonData = _reify_pact_body(
         body=pact_response_body,