diff --git a/nereid/nereid/api/api_v1/models/treatment_facility_models.py b/nereid/nereid/api/api_v1/models/treatment_facility_models.py
index b96d30be..eb301351 100644
--- a/nereid/nereid/api/api_v1/models/treatment_facility_models.py
+++ b/nereid/nereid/api/api_v1/models/treatment_facility_models.py
@@ -148,9 +148,7 @@ class DryWellFacility(FacilityBase):
     _constructor: str = "dry_well_facility_constructor"
 
 
-class DryWellFacilityFlowOrVolume(FacilityBase):
-    total_volume_cuft: float
-    treatment_rate_cfs: float
+class DryWellFacilityFlowOrVolume(FlowFacility, DryWellFacility):
     _constructor: str = "dry_well_facility_flow_or_volume_constructor"