Amending a return due to corrected Unemployment Comp

I received unemployment in 2020. Filed my claim before 15 March, but IRS hasn't corrected it yet (my taxes were complicated, self-employed, withdrawal from 401k, etc.). A month later I received a corrected 1099G from the state for the unemployment (Approx. $2,000 more). Should I wait for the IRS to first complete their correction before filing an amended return?