[cosmic_rays] Fix squeezing bug for nsets=1

6 jobs for master in 6 minutes and 19 seconds (queued for 3 seconds)