fixed on-shell final state momenta bug