Box 16 of the IRS form 1099-R would likely be the same as box 2a unless the income is assigned over more than one state income tax return.
Does entering the amount in box 2a clear the issue?