Optimizing sample_reassignment