First off download the instruction for the form from the IRS.  It will give you more clarification.  As you complete the form you will see how excludes the penalty.  However they need to realize that they are having to repay the distributions in thirds starting with the 2020 tax return.  They will still be subject to income tax on the portion they claim and repay for each year.