Minor fix to momentum corrections