Correction of the momentum program input