diff --git a/src/codegen/sdk/codebase/span.py b/src/codegen/sdk/codebase/span.py
index 12c3bb1b6..d8e7375cb 100644
--- a/src/codegen/sdk/codebase/span.py
+++ b/src/codegen/sdk/codebase/span.py
@@ -1,8 +1,9 @@
 from typing import Annotated, Any
 
-from pydantic import BaseModel, ConfigDict, WithJsonSchema
+from pydantic import BaseModel
+from pydantic.config import ConfigDict
 from pydantic.functional_validators import BeforeValidator
-from pydantic.json_schema import JsonSchemaValue
+from pydantic.json_schema import JsonSchemaValue, WithJsonSchema
 from pydantic_core.core_schema import ValidationInfo
 from tree_sitter import Point, Range