fix output in the routine that removes transverse momenta