diff --git a/python/valis/db/models.py b/python/valis/db/models.py
index 20f6627..f319c65 100644
--- a/python/valis/db/models.py
+++ b/python/valis/db/models.py
@@ -7,7 +7,7 @@
 import datetime
 import math
 from typing import Optional, Annotated, Any, TypeVar
-from pydantic import ConfigDict, BaseModel, Field, BeforeValidator, field_serializer, field_validator, FieldValidationInfo
+from pydantic import ConfigDict, BaseModel, Field, BeforeValidator, FieldSerializationInfo, field_serializer, field_validator, FieldValidationInfo
 from enum import Enum