diff --git a/cdr_schemas/cdr_responses/prospectivity.py b/cdr_schemas/cdr_responses/prospectivity.py
index 0a6d24a..138cfd5 100644
--- a/cdr_schemas/cdr_responses/prospectivity.py
+++ b/cdr_schemas/cdr_responses/prospectivity.py
@@ -21,9 +21,6 @@ class DataSource(CreateDataSource):
 
 
 class CreateProcessDataLayer(BaseModel):
-    cma: CriticalMineralAssessment = Field(
-        description="CMA with all information needed for processing"
-    )
     data_source: DataSource = Field(description="Data source to create this layer")
     title: str = Field(description="Title to use for processed layer")
     transform_methods: TranformMethods = Field(
@@ -39,5 +36,6 @@ class ProspectModelMetaData(BaseModel):
 
     model_run_id: str = Field(description="CDR id of the model run")
     cma: CriticalMineralAssessment = Field(description="CMA info")
+    model_type: str
     train_config: Union[SOMTrainConfig, NeuralNetUserOptions]
     evidence_layers: List[CreateProcessDataLayer]
diff --git a/cdr_schemas/prospectivity_input.py b/cdr_schemas/prospectivity_input.py
index a5e7693..b2e39fb 100644
--- a/cdr_schemas/prospectivity_input.py
+++ b/cdr_schemas/prospectivity_input.py
@@ -131,6 +131,7 @@ class CreateProspectModelMetaData(BaseModel):
     author: str = ""
     date: str = ""
     organization: str = ""
+    model_type: str
     train_config: Union[SOMTrainConfig, NeuralNetUserOptions]
     evidence_layers: List[DefineProcessDataLayer] = Field(
         description="Datasource and preprocess steps"
diff --git a/pyproject.toml b/pyproject.toml
index 335b41c..e7850af 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "cdr_schemas"
-version = "0.4.0"
+version = "0.4.1"
 description = "CDR Schemas"
 authors = []
 readme = "README.md"