identified a problem with the new collapse_f i think