The question is correct
The penalty is the same whether the shortfall is in a 401k, an IRA or both.
When considering your waiver request, the IRS will want to see in your explanation that you have made up the shortfall in a timely manner from the proper account(s).
@rosm2004
