Change API to enable samplers