You could directly correct the 1099-R form. In forms mode, see line 5 for federal withholding and lines 12 and 13 for the state.
Even if you chose to not enter the data in forms mode, you could look at the form to see what got in there during the interview.
