fix treatment selection