Some small corrections to avoid infinite loops at high momenta.