diff --git a/src/aeros_simulation/schema.py b/src/aeros_simulation/schema.py
index 99b8032..86e2b82 100644
--- a/src/aeros_simulation/schema.py
+++ b/src/aeros_simulation/schema.py
@@ -26,7 +26,7 @@ class SimulationNode(BaseModel):
     structure_name: Optional[str]
     schematic_name: Optional[str]
     schematic_id: Optional[UUID]
-    model_image: Optional[str]
+    model_image: Optional[list] = Field(None)
     equipment:Optional[MasterEquipment] = None
 
 class SimulationCalc(BaseModel):