fixed cohort bug