Yes, you are correct. Thank you for sharing your experience in the community. The Adjusted Gross Income (AGI) should be used from the original return.
For anyone else reading this and needing more help troubleshooting AGI issues, see:
What do I use for my AGI if I filed late, amended, or got a notice changing my return?
@yer_sister
