You must use the refund amount from the Originally filed return and not the corrected one the IRS will be doing for you.
