diff --git a/malariagen_data/anopheles.py b/malariagen_data/anopheles.py
index f7506c79e..142d3a2cd 100644
--- a/malariagen_data/anopheles.py
+++ b/malariagen_data/anopheles.py
@@ -526,6 +526,8 @@ def diversity_stats(
         cohorts: base_params.cohorts,
         cohort_size: base_params.cohort_size,
         region: base_params.regions,
+        min_cohort_size: Optional[base_params.min_cohort_size] = None,
+        max_cohort_size: Optional[base_params.max_cohort_size] = None,
         site_mask: Optional[base_params.site_mask] = base_params.DEFAULT,
         site_class: Optional[base_params.site_class] = None,
         sample_query: Optional[base_params.sample_query] = None,
@@ -544,7 +546,7 @@ def diversity_stats(
             sample_query=sample_query,
             sample_query_options=sample_query_options,
             cohort_size=cohort_size,
-            min_cohort_size=None,
+            min_cohort_size=min_cohort_size,
         )
 
         # Compute diversity stats for cohorts.
@@ -554,6 +556,8 @@ def diversity_stats(
                 cohort=(cohort_label, cohort_query),
                 cohort_size=cohort_size,
                 region=region,
+                min_cohort_size=min_cohort_size,
+                max_cohort_size=max_cohort_size,
                 site_mask=site_mask,
                 site_class=site_class,
                 sample_sets=sample_sets,