After you file

Agreed ... if your ONLY error was the stimulus amount then the IRS will/should automatically correct that "math error" and adjust the refund accordingly.   If for any reason they do not then and only then can/should you amend the return.  This can/may cause a delay in processing so just sit tight.