I would wait. You would not have to file an amended return in any case, just go to www.irs.gov/payments and make an additional $400 payment to your account. I would wait to see what happens. You should not be charged any late fees or penalties as long as the IRS receives the full amount by April 15 (either by correcting the check or you making an extra payment).
