Fix bug momentum index