After you file

i would not file an amended Federal return. you properly claimed it on your 1040. as a matter of fact you should put it back. if you don't this could affect your 2022 return when you do the rollover/transfer when starting that year.