fix 2nd bug in momemtum