After you file

Check all your transcripts to see if that was already handled but if it was not then amending the return to capture the missing payment is the correct way to go.