That is correct if you have two 1099R's from the 401K, if you have only one 1099R you will have to do what is listed below.
-------------------------------------
You will actually have to split this out and show it as two separate 1099R's one for the Roth amount only and one for the Traditional IRA amount.
Form 1099-R -- Distributions from Pensions, Annuities, Retirement or Profit-Sharing Plans, IRAs, Insurance Contracts, etc.: A single Form 1099-R rolled into multiple types of retirement plans (IRA, Roth IRA, other qualified retirement plan, etc.) is not supported. In this situation, you must determine the amount rolled into each type of plan. Then treat each part of the rollover as a separate distribution and enter on multiple 1099-R worksheets. "
Form 1099-R -- Distributions from Pensions, Annuities, Retirement or Profit-Sharing Plans, IRAs, Insurance Contracts, etc.: A single Form 1099-R rolled into multiple types of retirement plans (IRA, Roth IRA, other qualified retirement plan, etc.) is not supported. In this situation you must determine the amount rolled into each type of plan. Then treat each part of the rollover as a separate distribution and enter on multiple 1099-R worksheets. "
Source #27 https://digitalasset.intuit.com/DOCUMENT/A5pxS2bpm/Unsupported-Calcs-GEN12868.pdf
