diff --git a/simpeg_drivers/options.py b/simpeg_drivers/options.py
index db237ef9..27acae12 100644
--- a/simpeg_drivers/options.py
+++ b/simpeg_drivers/options.py
@@ -22,6 +22,7 @@
     BooleanData,
     DataAssociationEnum,
     FloatData,
+    IntegerData,
     NumericData,
     ReferencedData,
 )
@@ -55,7 +56,7 @@ class ActiveCellsOptions(BaseModel):
     )
     topography_object: Points | None = None
     topography: FloatData | float | None = None
-    active_model: BooleanData | None = None
+    active_model: BooleanData | IntegerData | None = None
 
     @model_validator(mode="before")
     @classmethod