Took 401K) Distribution for Home Purchase, then replaced it 60 days later with $10K Less

In 2017, I withdrew $150K from my 401(K) to close the purchase of a home.  About 50 days later, I replaced all but $10K with a rollover back to the same account.  For some reason, I thought Turbo Tax would account for this, but not so, and now I must amend my return to reflect that $10K of income unreported.  The IRS recognizes both the distribution and the rollover, so its a matter of inserting a number in the right place.  Where is that?