Turbo tax figured my Stimulus wrong - IRS fixed it - my amended return is using the original wrong stimulus amount

On Feb 12, the stimulus questions were "did you get both $600 payments."  I got two payments, so I said yes, not  even thinking the first stimulus was 1200 and not 600.  Turbo tax said I was entitled to more, and added $600 to my refund  - making it  $1877.  I filed on the 13th. Turbo tax has subsequently changed their questions and their calculations, and when I look at the return in Turbo Tax now, my refund shows (correctly) it would be $1277.  The IRS corrected the $600 error and refunded  $1277.   I  then got a surprise K1 and need to do an amended return - but Turbo Tax is basing the tax on the original $1877 and not the amount the IRS refunded (and which Turbo tax now shows as the tax refund due).   How do I get the amended return to calculate on the corrected Turbo Tax return?