diff --git a/src/codegen/agents/client/openapi_client/api_response.py b/src/codegen/agents/client/openapi_client/api_response.py
index e8af61447..3842a95da 100644
--- a/src/codegen/agents/client/openapi_client/api_response.py
+++ b/src/codegen/agents/client/openapi_client/api_response.py
@@ -2,13 +2,10 @@
 
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Generic, TypeVar
+from typing import Generic, TypeVar
 
 from pydantic import BaseModel, Field, StrictBytes, StrictInt
 
-if TYPE_CHECKING:
-    from collections.abc import Mapping
-
 T = TypeVar("T")
 
 
@@ -16,7 +13,7 @@ class ApiResponse(BaseModel, Generic[T]):
     """API response object"""
 
     status_code: StrictInt = Field(description="HTTP status code")
-    headers: Mapping[str, str] | None = Field(None, description="HTTP headers")
+    headers: dict[str, str] | None = Field(None, description="HTTP headers")
     data: T = Field(description="Deserialized data given the data type")
     raw_data: StrictBytes = Field(description="Raw data (HTTP response body)")