diff --git a/api/app/models/model_dto.py b/api/app/models/model_dto.py
index 6358026e..2229cf8b 100644
--- a/api/app/models/model_dto.py
+++ b/api/app/models/model_dto.py
@@ -86,6 +86,7 @@ class ModelFeatures(BaseModel):
 
 
 class ModelIn(BaseModel, validate_assignment=True):
+    uuid: Optional[UUID] = None
     name: str
     description: Optional[str] = None
     model_type: ModelType
@@ -179,6 +180,7 @@ def timestamp_must_be_datetime(self) -> Self:
     def to_model(self) -> Model:
         now = datetime.datetime.now(tz=datetime.UTC)
         return Model(
+            uuid=self.uuid,
             name=self.name,
             description=self.description,
             model_type=self.model_type.value,