Is it possible to somehow modify the sampling from the model to account for that?